Update app.py

app.py

@@ -292,16 +292,17 @@ with gr.Blocks(title="Texture Super-Resolution") as demo:
                     )
                     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
+                streaming=False,
+                show_download_button=True
             )
 
+
     # Add examples
     gr.Examples(
         examples=EXAMPLES,
@@ -353,8 +354,7 @@ 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"]
+    file_types=["png"]
 )
 
 if __name__ == "__main__":