Update app.py

app.py

@@ -83,6 +83,11 @@ def process_img (image, prompt):
     restored_img = (restored_img * 255.0).round().astype(np.uint8)  # float32 to uint8
     return Image.fromarray(restored_img) #(image, Image.fromarray(restored_img))
 
+    output_image = Image.fromarray(restored_img)
+    output_image.format = "PNG"
+
+    return output_image
+
 
 
 title = "InstructIR ✏️🖼️ 🤗"
@@ -149,10 +154,10 @@ css = """
 demo = gr.Interface(
     fn=process_img,
     inputs=[
-            gr.Image(type="pil", label="Input", value="images/a4960.jpg"),
-            gr.Text(label="Prompt", value="my colors are too off, make it pop so I can use it in instagram")
+        gr.Image(type="pil", label="Input", value="images/a4960.jpg"),
+        gr.Text(label="Prompt", value="my colors are too off, make it pop so I can use it in instagram")
     ],
-    outputs=[gr.Image(type="pil", label="Ouput")],
+    outputs=[gr.Image(type="pil", image_mode="RGB", format="png", label="Output")],
     title=title,
     description=description,
     article=article,