Update app.py

app.py

@@ -278,7 +278,29 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
                 submit_btn = gr.Button("Submit", variant="primary")
 
         with gr.Column():
-            output_image = gr.Image(label="Generated Image", type="pil", format="png")
+            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",
+                elem_id="output_texture",
+                tool="custom-download",
+                download_fn=custom_save_image
+            )
 
     # Add examples
     gr.Examples(
@@ -328,5 +350,12 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
         ]
     )
 
+demo.config = gr.Config(
+    png_quality=100,
+    png_compression=0,
+    file_types=["png"],
+    default_enabled_tools=["download"]
+)
+
 if __name__ == "__main__":
     demo.launch()