Fixes

app.py

@@ -32,7 +32,10 @@ def generate(prompt, option, seed):
     pipe = pipes[option]
     set_seed(int(seed))
     start_time = time.time()
-    output = pipe(prompt, num_inference_steps=50, output_type="pil", height=512, width=512)
+    if "Torch" in option:
+        output = pipe(prompt, num_inference_steps=50, output_type="pil", height=512, width=512)
+    else:
+        output = pipe(prompt, num_inference_steps=50, output_type="pil")
     elapsed_time = time.time() - start_time
     return (output.images[0], "{:10.4f}".format(elapsed_time))
 
@@ -44,9 +47,9 @@ model_options = [option for option in pipes.keys()]
 
 gr.Interface(
     fn=generate,
-    inputs=[gr.inputs.Textbox(placeholder="cartoon bird", label="Prompt", lines=1),
+    inputs=[gr.inputs.Textbox(default="cartoon bird", label="Prompt", lines=1),
             gr.inputs.Dropdown(choices=model_options, default=model_options[-1], label="Model version"),
-            gr.inputs.Textbox(placeholder="42", label="Seed", lines=1)
+            gr.inputs.Textbox(default="42", label="Seed", lines=1)
            ],
     outputs=[gr.outputs.Image(type="pil", label="Generated Image"), gr.outputs.Textbox(label="Inference time")],
     title="OpenVINO-optimized Stable Diffusion",