Update app.py

app.py

@@ -6,7 +6,7 @@ pipe = StableDiffusionPipeline.from_pretrained("paavansundar/image-diffusion", t
 def generate_img(prompt):
     pipe.enable_attention_slicing()
     #pipe(prompt).images[0]
-    return success
+    return pipe(prompt).images[0]
 with gr.Blocks() as demo:
     im = gr.Image()
     prompt = gr.Textbox(label="Input", lines=2)