Update app.py

app.py

@@ -3,6 +3,7 @@ from gradio_client import Client, handle_file
 from PIL import Image
 from io import BytesIO
 import os
+import tempfile
 
 def upscale_image(url):
     client = Client("doevent/Face-Real-ESRGAN")
@@ -13,21 +14,24 @@ def upscale_image(url):
     )
     # print("\nTask Completed!")
     # return result
+    # Read the image from the file path returned by the model
     if os.path.exists(result):
         with open(result, 'rb') as img_file:
             img_data = img_file.read()
-    
+
         # 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
+        # Save the converted image to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
+            img.save(temp_file, format="PNG")
+            temp_file_path = temp_file.name
+
+        # Optionally, delete the temp file after processing (you can remove this line if not needed)
         os.remove(result)
-    
+
         print("\nTask Completed!")
-        return output
+        return temp_file_path
 
 
 app = gr.Interface(upscale_image,