Update app.py

app.py

@@ -25,14 +25,14 @@ def run_theia(image, pred_iou_thresh, stability_score_thresh):
         "LiheYoung/depth-anything-large-hf",
     ]
     feature_means, feature_vars = load_feature_stats(target_model_names, stat_file_root="feature_stats")
-    
+
     mask_generator, sam_model = prepare_mask_generator('cuda')
     depth_anything_model_name = "LiheYoung/depth-anything-large-hf"
     depth_anything_decoder, _ = prepare_depth_decoder(depth_anything_model_name, 'cuda')
 
     image = torchvision.transforms.Resize(size=(224, 224))(image)
     images = [image]
-    
+
     theia_decode_results, gt_results = decode_everything(
         theia_model=theia_model,
         feature_means=feature_means,
@@ -70,10 +70,13 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(label="Input Image", type="pil")
-            pred_iou_thresh = gr.Slider(0.05, 0.95, step=0.05, value=0.5, label="SAM Pred IoU Thresh")
-            stability_score_thresh = gr.Slider(0.05, 0.95, step=0.05, value=0.7, label="SAM Stability Score Thresh")
+
+            with gr.Accordion("Advanced Settings", open=False):
+                pred_iou_thresh = gr.Slider(0.05, 0.95, step=0.05, value=0.5, label="SAM Pred IoU Thresh")
+                stability_score_thresh = gr.Slider(0.05, 0.95, step=0.05, value=0.7, label="SAM Stability Score Thresh")
+
             submit_button = gr.Button("Submit")
-        
+
         with gr.Column():
             dinov2_output = gr.Gallery(label="DINOv2", type="numpy")
             sam_output = gr.Gallery(label="SAM", type="numpy")