Update app.py

app.py

@@ -6,11 +6,17 @@ from diffusers import DDPMScheduler, StableDiffusionPipeline
 import time
 
 def create_pipeline(name):
-    if name == "svjack/Stable-Diffusion-Pokemon-en":
+    if name == "valhalla/sd-pokemon-model":
         scheduler = DDPMScheduler(beta_start=0.00085, beta_end=0.012,
              beta_schedule="scaled_linear", num_train_timesteps=1000)
         pipe = StableDiffusionPipeline.from_pretrained(name, scheduler=scheduler)
         pipe.safety_checker = lambda images, clip_input: (images, False)
+    elif name == "stable-diffusion-pokemons-valhalla-fp32":
+        scheduler = DDPMScheduler(beta_start=0.00085, beta_end=0.012,
+             beta_schedule="scaled_linear", num_train_timesteps=1000)
+        pipe = OVStableDiffusionPipeline.from_pretrained(name, compile=False, scheduler=scheduler)
+        pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
+        pipe.compile()
     else:
         pipe = OVStableDiffusionPipeline.from_pretrained(name, compile=False)
         pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)