Added torch model

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 from optimum.intel.openvino import OVStableDiffusionPipeline
 from diffusers.training_utils import set_seed
-from diffusers import LMSDiscreteScheduler
+from diffusers import LMSDiscreteScheduler, StableDiffusionPipeline
 
 import time
 
-scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012,
-                                 beta_schedule="scaled_linear", num_train_timesteps=1000)
-pipe_fp32 = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-pokemons-fp32", compile=False, scheduler=scheduler)
+# scheduler = DDPMScheduler(beta_start=0.00085, beta_end=0.012,
+#                                  beta_schedule="scaled_linear", num_train_timesteps=1000)
+pipe_torch_fp32 = StableDiffusionPipeline("svjack/Stable-Diffusion-Pokemon-en")
+
+pipe_fp32 = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diffusion-pokemons-fp32", compile=False) scheduler=scheduler)
 pipe_fp32.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
 pipe_fp32.compile()
 
@@ -22,9 +24,10 @@ pipe_tome_int8.compile()
 prompt = "cartoon bird"
 
 pipes = {
-    "FP32": pipe_fp32,
-    "8-bit quantized": pipe_int8,
-    "Merged and quantized": pipe_tome_int8
+    "Torch fp32"
+    "OpenVINO fp32": pipe_fp32,
+    "OpenVINO 8-bit quantized": pipe_int8,
+    "OpenVINO merged and quantized": pipe_tome_int8
 }
 
 def generate(image, option):
@@ -38,10 +41,12 @@ examples = ["cartoon bird",
             "a drawing of a green pokemon with red eyes",
             "plant pokemon in jungle"]
 
+choices = [option for option in pipes.keys()]
+
 gr.Interface(
     fn=generate,
     inputs=[gr.inputs.Textbox(placeholder="cartoon bird", label="Prompt", lines=1),
-            gr.inputs.Dropdown(choices=[option for option in pipes.keys()], default="Merged and quantized", label="Model version"),
+            gr.inputs.Dropdown(choices=choices], default=choices[-1], label="Model version"),
            ],
     outputs=[gr.outputs.Image(type="pil", label="Generated Image"), gr.outputs.Textbox(label="Inference time")],
     title="OpenVINO-optimized Stable Diffusion",