Fix: remove show_download_button (not in Gradio 6.x Image)

app.py

@@ -403,15 +403,15 @@ with gr.Blocks(title="UNIStainNet — Virtual IHC Staining") as demo:
 
             gr.HTML('<p class="section-heading">Input & Ground Truth</p>')
             with gr.Row():
-                gal_he = gr.Image(type="pil", label="H&E Input", height=280, show_download_button=False)
-                gal_gt = gr.Image(type="pil", label="Ground Truth IHC", height=280, show_download_button=False)
+                gal_he = gr.Image(type="pil", label="H&E Input", height=280)
+                gal_gt = gr.Image(type="pil", label="Ground Truth IHC", height=280)
 
             gr.HTML('<p class="section-heading">Generated IHC Stains (all from the same H&E)</p>')
             with gr.Row():
-                gal_her2 = gr.Image(type="pil", label="Generated HER2", height=280, show_download_button=False)
-                gal_ki67 = gr.Image(type="pil", label="Generated Ki67", height=280, show_download_button=False)
-                gal_er = gr.Image(type="pil", label="Generated ER", height=280, show_download_button=False)
-                gal_pr = gr.Image(type="pil", label="Generated PR", height=280, show_download_button=False)
+                gal_her2 = gr.Image(type="pil", label="Generated HER2", height=280)
+                gal_ki67 = gr.Image(type="pil", label="Generated Ki67", height=280)
+                gal_er = gr.Image(type="pil", label="Generated ER", height=280)
+                gal_pr = gr.Image(type="pil", label="Generated PR", height=280)
 
             def _show_gallery_wrapper(display_name):
                 return show_gallery(display_name, gallery_data, gallery_name_map)