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 = "/repository/checkpoints/sam"
+        self.model_checkpoint_path = os.path.abspath("/repository/checkpoints/sam")
 
         self.download_parameters()
         self.sam = build_sam(checkpoint=self.model_checkpoint_path).to(device)
@@ -813,9 +813,9 @@ 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):
-            print("我进来了！")
+            print("I'm in！")
             # wget.download(url,out=self.model_checkpoint_path)
-            wget.download(url,out=path)
+            wget.download(url,out=self.model_checkpoint_path)
 
     def show_mask(self, mask, ax, random_color=False):
         if random_color:
@@ -917,7 +917,7 @@ 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 = "repository/checkpoints/groundingdino"
+        self.model_checkpoint_path = os.path.abspath("repository/checkpoints/groundingdino")
         self.model_config_path = "repository/checkpoints/grounding_config.py"
         self.download_parameters()
         self.box_threshold = 0.3