Update app.py

app.py

@@ -3,7 +3,7 @@ from optimum.intel.openvino import OVStableDiffusionPipeline
 from diffusers.training_utils import set_seed
 
 quantized_pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/Stable-Diffusion-Pokemon-en-quantized", compile=False)
-quantized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
+quantized_pipe.reshape(batch_size=1, height=256, width=256, num_images_per_prompt=1)
 quantized_pipe.compile()
 
 prompt = "cartoon bird"
@@ -12,14 +12,19 @@ def generate(image):
     output = quantized_pipe(prompt, num_inference_steps=50, output_type="pil")
     return output.images[0]
 
+examples = ["cartoon bird",
+            "a drawing of a green pokemon with red eyes",
+            "plant pokemon in jungle"]
+
 gr.Interface(
     fn=generate,
     inputs=gr.inputs.Textbox(placeholder="cartoon bird", 
-                             label="Prompt"
-                             "query",
+                             label="Prompt",
                              lines=1),
     outputs=gr.outputs.Image(type="pil", label="Generated Image"),
-    title="Quantized stable diffusion",
+    title="OpenVINO-optimized Stable Diffusion",
     description="",
+    verbose=True,
+    examples=examples,
     theme="huggingface",
 ).launch()