Update app.py

app.py

@@ -1,4 +1,4 @@
-from PIL import Image
+import numpy as np
 import matplotlib.pyplot as plt
 from diffusers import DiffusionPipeline
 
@@ -9,13 +9,14 @@ pipeline = DiffusionPipeline.from_pretrained("ankush-003/retinal_fundus")
 def generate_image():
     image = pipeline().images[0]
     image.save("trial.png")
-    img = Image.open("trial.png")
-    # plt.imshow(img)
-    # plt.axis("off")
-    # plt.show()
+    img = plt.imread("trial.png")
+    # Display the image (optional)
+    plt.imshow(img)
+    plt.axis("off")
+    plt.show()
     return img
 
 # gradio interface
 import gradio as gr
-iface = gr.Interface(fn=generate_image, inputs=None, outputs="image")
+iface = gr.Interface(fn=generate_image, inputs=None, outputs=[gr.Image(label="Generated Image", type="numpy")])
 iface.launch(debug=True)