Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 import subprocess
 import torch
 from PIL import Image
-import numpy as np
 
 UPLOAD_FOLDER = 'uploads'
 OUTPUT_FOLDER = 'outputs'
@@ -36,7 +35,12 @@ def gradio_interface(image):
         if result.returncode != 0:
             return f"Error: {result.stderr}"
 
-        return Image.open(output_path)
+        # Check if output image was created
+        if not os.path.exists(output_path):
+            return "Error: Output image not generated."
+
+        # Explicitly reload the image
+        return Image.open(output_path).copy()
 
     except Exception as e:
         return f"Exception: {str(e)}"
@@ -49,4 +53,4 @@ iface = gr.Interface(
     title="Image Restoration with NAFNet"
 )
 
-iface.launch()  # Removed share=True
+iface.launch()