Merge branch 'main' into add_figs

Project-MONAI · Oct 24, 2024 · b82fcf7 · b82fcf7
2 parents 13563c9 + 788bd21
commit b82fcf7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/m3/demo/experts/expert_monai_vista3d.py b/m3/demo/experts/expert_monai_vista3d.py
@@ -184,12 +184,12 @@ def run(
             raise ValueError(f"Label group {arg_matches[0]} is not accepted by the VISTA-3D model.")
 
         if arg_matches[0] != "everything":
-            vista3d_prompts = {"classes": list(label_groups[arg_matches[0]].keys())}
+            vista3d_prompts = [cls_idx for _, cls_idx in label_groups[arg_matches[0]].items()]
 
         # Trigger the VISTA-3D model
         input_dict = {"image": img_file}
         if vista3d_prompts is not None:
-            input_dict["prompts"] = vista3d_prompts
+            input_dict["label_prompt"] = vista3d_prompts
 
         sys.path = [self.bundle_root] + sys.path
 

diff --git a/m3/demo/gradio_monai_vila2d.py → m3/demo/gradio_m3.py b/m3/demo/gradio_monai_vila2d.py → m3/demo/gradio_m3.py