Hide info testboxes

app.py

@@ -13,6 +13,8 @@ from src.smc.inference import (
     FTInferenceConfig,
 )
 
+GALLERY_HEIGHT = "224px"
+
 def get_device():
     if not hasattr(get_device, "last_allocated"):
         get_device.last_allocated = -1 # type: ignore
@@ -162,10 +164,10 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=2):
             pretrained_gallery = gr.Gallery(
-                label="Pretrained model outputs", show_label=True, elem_id="pretrained_gallery", height="240px", columns=4,
+                label="Pretrained model outputs", show_label=True, elem_id="pretrained_gallery", height=GALLERY_HEIGHT, columns=4,
                 object_fit="contain",
             )
-            pretrained_info = gr.Textbox(label="Pretrained info", interactive=False)
+            pretrained_info = gr.Textbox(label="Pretrained info", interactive=False, visible=False)
 
     # --- SMC-grad method row ---
     with gr.Row():
@@ -205,10 +207,10 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=2):
             smc_grad_gallery = gr.Gallery(
-                label="SMC-grad outputs", show_label=True, elem_id="smc_grad_gallery", height="240px", columns=4,
+                label="SMC-grad outputs", show_label=True, elem_id="smc_grad_gallery", height=GALLERY_HEIGHT, columns=4,
                 object_fit="contain",
             )
-            smc_grad_info = gr.Textbox(label="SMC-grad info", interactive=False)
+            smc_grad_info = gr.Textbox(label="SMC-grad info", interactive=False, visible=False)
             
     # --- FT method row ---
     with gr.Row():
@@ -222,10 +224,10 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=2):
             ft_gallery = gr.Gallery(
-                label="Finetuned model outputs", show_label=True, elem_id="ft_gallery", height="240px", columns=4,
+                label="Finetuned model outputs", show_label=True, elem_id="ft_gallery", height=GALLERY_HEIGHT, columns=4,
                 object_fit="contain",
             )
-            ft_info = gr.Textbox(label="Finetuned info", interactive=False)
+            ft_info = gr.Textbox(label="Finetuned info", interactive=False, visible=False)
 
     # --- Wiring ---
     # 1) Quick 'running' update attached to the button so the UI shows immediate feedback.