Update app.py

app.py

@@ -68,11 +68,12 @@ def server(input: Inputs, output: Outputs, session: Session):
         
         return original_file.name, combined_file.name
     
-    def image_to_base64(image):
-        buffered = BytesIO()
-        image.save(buffered, format="PNG")
-        img_str = base64.b64encode(buffered.getvalue()).decode()
-        return f"data:image/png;base64,{img_str}"
+        combined_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+        original_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+        image.save(original_file.name, "PNG", quality=100)
+        mask.save(combined_file.name, "PNG", quality=100)
+        
+        return original_file.name, combined_file.name
 
     @render.image
     def original_image():
@@ -88,7 +89,6 @@ def server(input: Inputs, output: Outputs, session: Session):
         if result is None:
             return None
         _, img_path = result
-        image = Image.open(img_path)
-        return {"src": image_to_base64(image), "width": "300px"}
+        return {"src": img_path, "width": "300px"}
 
 app = App(app_ui, server)