Update src/app.py

src/app.py

@@ -136,14 +136,7 @@ def process(
         solver_type=solver_type,
     )
     
-    output_path = "/tmp/enhanced_image.png"
-    enhanced_image.save(output_path, format="PNG")
-
-    # Load the saved PNG image to ensure it is returned in PNG format
-    enhanced_image_png = Image.open(output_path)
-
-    return output_path, enhanced_image_png
-    #return enhanced_image
+    return enhanced_image
 
 
 with gr.Blocks() as demo: