try gradio==4.38.1

app.py

@@ -93,7 +93,21 @@ generated_object_map = {}
 
 # Prepare models
 ## Grounding SAM
-sam2_predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large", device=torch.device("cpu"))
+if not os.path.exists("./checkpoints/sam2.1_hiera_large.pt"):
+    os.makedirs("./checkpoints/", exist_ok=True)
+    os.system("cd ./checkpoints/")
+    os.system("wget https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt")
+    os.system("cd ..")
+SAM2_CHECKPOINT = "./checkpoints/sam2.1_hiera_large.pt"
+SAM2_CONFIG = "configs/sam2.1/sam2.1_hiera_l.yaml"
+
+sam2_model = build_sam2(
+    config_file=SAM2_CONFIG,
+    ckpt_path=SAM2_CHECKPOINT,
+    device=torch.device("cpu"),
+)
+sam2_predictor = SAM2ImagePredictor(sam2_model)
+# sam2_predictor = SAM2ImagePredictor.from_pretrained("facebook/sam2-hiera-large", device=torch.device("cpu"))
 
 ############## 3D-Fixer model
 model_dir = 'HorizonRobotics/3D-Fixer'

requirements.txt

@@ -28,11 +28,12 @@ icecream
 plyfile
 pycocotools
 shapely
-gradio==4.44.1
 gradio_image_prompter
 spconv-cu118
 huggingface_hub==0.36.0
 git+https://github.com/EasternJournalist/utils3d.git@9a4eb15e4021b67b12c460c7057d642626897ec8
-gradio_client==1.3.0
 pydantic==2.10.6
-fastapi==0.112.2
+gradio==4.38.1
+gradio_client==1.1.0
+fastapi==0.111.1
+httpx==0.27.0