Update visual_chatgpt.py

visual_chatgpt.py

@@ -807,14 +807,16 @@ class Segmenting:
     def download_parameters(self):
         ##############
         print("getcwd",os.getcwd())
-        print("ls-R",os.system("ls -R"))
-        print("find",os.system("find ./ -name 'handler.py'"))
-        print("finddir",os.system("find ./ -type d -iname 'checkpoints'"))
+        REPO_DIR = os.path.dirname(__file__)
+        path=os.path.join(REPO_DIR, "checkpoints/sam")
+        # print("ls-R",os.system("ls -R"))
+        # print("find",os.system("find ./ -name 'handler.py'"))
+        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("/repository/checkpoints"):
-            os.makedirs("/repository/checkpoints")
+        if not os.path.exists(path):
+            os.makedirs(path)
             # wget.download(url,out=self.model_checkpoint_path)
-            wget.download(url,out="/repository/checkpoints/sam")
+            wget.download(url,out=path)
 
     def show_mask(self, mask, ax, random_color=False):
         if random_color: