Update app.py

app.py

@@ -24,15 +24,15 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
 
 def generate_image(text):
 
-    image = pipe("krishna", num_inference_steps=2, guidance_scale=0).images[0]
+    pipe("krishna", num_inference_steps=2, guidance_scale=0).images[0].save("output.png")
     
-    return image
+    return "output.png"
 
 # Create a Gradio interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(lines=5, label="Enter a description for the image"),
-    outputs="image",
+    outputs=outputs=gr.outputs.Image(type="filepath", label="Generated Image"),
     title="Text to Image Generation",
     description="Enter a text description and get an image.",
     theme="compact"