Update visual_chatgpt.py

visual_chatgpt.py

@@ -797,7 +797,7 @@ class Segmenting:
         print(f"Inintializing Segmentation to {device}")
         self.device = device
         self.torch_dtype = torch.float16 if 'cuda' in device else torch.float32
-        self.model_checkpoint_path = os.path.join("checkpoints","sam")
+        self.model_checkpoint_path = "/repository/checkpoints/sam"
 
         self.download_parameters()
         self.sam = build_sam(checkpoint=self.model_checkpoint_path).to(device)
@@ -813,7 +813,6 @@ class Segmenting:
         print("finddir",os.system("find /repository -type d -iname 'checkpoints'"))
         url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
         if not os.path.exists(path):
-            os.makedirs(path)
             print("我进来了！")
             # wget.download(url,out=self.model_checkpoint_path)
             wget.download(url,out=path)
@@ -918,8 +917,8 @@ class Text2Box:
         print(f"Initializing ObjectDetection to {device}")
         self.device = device
         self.torch_dtype = torch.float16 if 'cuda' in device else torch.float32
-        self.model_checkpoint_path = os.path.join("checkpoints","groundingdino")
-        self.model_config_path = os.path.join("checkpoints","grounding_config.py")
+        self.model_checkpoint_path = "repository/checkpoints/groundingdino"
+        self.model_config_path = "repository/checkpoints/grounding_config.py"
         self.download_parameters()
         self.box_threshold = 0.3
         self.text_threshold = 0.25