Update src/app.py

src/app.py

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