Update app.py

app.py

@@ -10,12 +10,12 @@ import time
 
 
 def create_pipeline(name):
-    if name == "valhalla/sd-pokemon-model":
+    if name == "svjack/Stable-Diffusion-Pokemon-en": #"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":
+    elif name == "OpenVINO/stable-diffusion-pokemons-fp32": #"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)
@@ -28,10 +28,10 @@ def create_pipeline(name):
     return pipe
 
 pipes = {
-    "Torch fp32": "valhalla/sd-pokemon-model", #"svjack/Stable-Diffusion-Pokemon-en",
-    "OpenVINO fp32": "OpenVINO/stable-diffusion-pokemons-valhalla-fp32", #"OpenVINO/stable-diffusion-pokemons-fp32",
-    "OpenVINO 8-bit quantized": "OpenVINO/stable-diffusion-pokemons-valhalla-quantized-agressive", #"OpenVINO/stable-diffusion-pokemons-quantized-aggressive",
-    "OpenVINO merged and quantized": "OpenVINO/stable-diffusion-pokemons-valhalla-tome-quantized-agressive", #"OpenVINO/stable-diffusion-pokemons-tome-quantized-aggressive"
+    "Torch fp32": "svjack/Stable-Diffusion-Pokemon-en", #"valhalla/sd-pokemon-model"
+    "OpenVINO fp32": "OpenVINO/stable-diffusion-pokemons-fp32", #"OpenVINO/stable-diffusion-pokemons-valhalla-fp32"
+    "OpenVINO 8-bit quantized": "OpenVINO/stable-diffusion-pokemons-quantized-aggressive", #"OpenVINO/stable-diffusion-pokemons-valhalla-quantized-agressive"
+    "OpenVINO merged and quantized": "OpenVINO/stable-diffusion-pokemons-tome-quantized-aggressive" #"OpenVINO/stable-diffusion-pokemons-valhalla-tome-quantized-agressive"
 }
 
 # prefetch pipelines on start