Update app.py

app.py

@@ -278,20 +278,12 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
                 submit_btn = gr.Button("Submit", variant="primary")
 
         with gr.Column():
-            def custom_save_image(image):
-                """Custom save function to ensure maximum PNG quality"""
-                if isinstance(image, Image.Image):
-                    import io
-                    buffer = io.BytesIO()
-                    image.save(
-                        buffer, 
-                        format='PNG',
-                        optimize=False,
-                        quality=100,
-                        subsampling=0
-                    )
-                    buffer.seek(0)
-                    return buffer
+            output_image = gr.Image(
+                label="Generated Image", 
+                type="pil",
+                format="png",
+                show_download_button=True
+            )
         
             output_image = gr.Image(
                 label="Generated Image",