Update app.py

app.py

@@ -21,8 +21,6 @@ pipe_tome_int8 = OVStableDiffusionPipeline.from_pretrained("OpenVINO/stable-diff
 pipe_tome_int8.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
 pipe_tome_int8.compile()
 
-prompt = "cartoon bird"
-
 pipes = {
     "Torch fp32"
     "OpenVINO fp32": pipe_fp32,
@@ -30,7 +28,7 @@ pipes = {
     "OpenVINO merged and quantized": pipe_tome_int8
 }
 
-def generate(image, option):
+def generate(prompt, option):
     pipe = pipes[option]
     start_time = time.time()
     output = pipe(prompt, num_inference_steps=50, output_type="pil")