download

app.py

@@ -58,6 +58,10 @@ CUSTOM_CSS = """
 .gradio-container, .gradio-container * {
     font-family: 'Arial', 'Helvetica', 'Microsoft YaHei', '微软雅黑', sans-serif !important;
 }
+button[aria-label="Download"],
+a[download] {
+    display: none !important;
+}
 """
 
 # --- Base directory for all models ---
@@ -130,7 +134,7 @@ def generate_colorbar_preview(color_name):
             rgb[..., 2] = high_half
         elif color_name == "Cyan Hot":
             rgb[..., 0] = high_half
-            rgb[..., 1] = low_half
+            rgb[..., 1] = high_half
             rgb[..., 2] = low_half
         elif color_name == "Magenta Hot":
             rgb[..., 0] = low_half
@@ -606,7 +610,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
                     t2i_gallery_out = gr.Gallery(label="Generated Images", columns=3, height="auto")
                     with gr.Row(equal_height=True):
                         with gr.Column(scale=2):
-                            t2i_color = gr.Dropdown(choices=COLOR_MAPS, value="Green (GFP)", label="Pseudocolor (Adjust after generation)")
+                            t2i_color = gr.Dropdown(choices=COLOR_MAPS, value="Green (GFP)", label="Pseudocolor (Adjust after generation)", show_download_button=False)
                         with gr.Column(scale=2):
                             t2i_colorbar = gr.Image(label="Colorbar", show_label=False, container=False, height=40, show_download_button=False, interactive=False)
                         with gr.Column(scale=1):
@@ -700,7 +704,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
                     m2i_seed = gr.Number(label="Seed", value=42)
                     m2i_btn = gr.Button("Generate", variant="primary")
                 with gr.Column(scale=2):
-                    m2i_gal_out = gr.Gallery(label="Generated Samples", columns=5, height="auto")
+                    m2i_gal_out = gr.Gallery(label="Generated Samples", columns=5, height="auto", show_download_button=False)
                     with gr.Row(equal_height=True):
                         with gr.Column(scale=2):
                             m2i_color = gr.Dropdown(choices=COLOR_MAPS, value="Grayscale", label="Pseudocolor")