Update app.py

app.py

@@ -26,6 +26,6 @@ def generate(prompt):
 # def greet(prompt):
 #     image = pipe(prompt, num_inference_steps=1, guidance_scale=0).images[0].save("output.png")
 #     return image
-
-demo = gr.Interface(fn=generate, inputs="text", outputs="image")
+output_image = gr.outputs.Image(type="PIL")
+demo = gr.Interface(fn=generate, inputs="text", outputs=output_image)
 demo.launch()