Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from gradio_client import Client, handle_file
 from PIL import Image
 from io import BytesIO
+import os
 
 def upscale_image(url):
     client = Client("doevent/Face-Real-ESRGAN")
@@ -12,13 +13,21 @@ def upscale_image(url):
     )
     # print("\nTask Completed!")
     # return result
-    img = Image.open(BytesIO(result))
-    output = BytesIO()
-    img.save(output, format="PNG")
-    output.seek(0)
+    if os.path.exists(result):
+        with open(result, 'rb') as img_file:
+            img_data = img_file.read()
     
-    print("\nTask Completed!")
-    return output
+        # Convert result to PNG
+        img = Image.open(BytesIO(img_data))
+        output = BytesIO()
+        img.save(output, format="PNG")
+        output.seek(0)
+        
+        # Optionally, delete the temp file after processing
+        os.remove(result)
+    
+        print("\nTask Completed!")
+        return output
 
 
 app = gr.Interface(upscale_image,