Zhibinhong
/

ReplacerRemover

Inference Endpoints

Model card Files Files and versions Community

Zhibinhong commited on Apr 27, 2023

Commit

212790e

•

1 Parent(s): 644fc6c

Update visual_chatgpt.py

Files changed (1) hide show

visual_chatgpt.py +1 -1

visual_chatgpt.py CHANGED Viewed

@@ -807,7 +807,7 @@ class Segmenting:
     def download_parameters(self):
         url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
         if not os.path.exists(self.model_checkpoint_path):
-            wget.download(url,out="/repository/"+self.model_checkpoint_path)
     def show_mask(self, mask, ax, random_color=False):
         if random_color:

     def download_parameters(self):
         url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
         if not os.path.exists(self.model_checkpoint_path):
+            wget.download(url,out="repository/"+self.model_checkpoint_path)
     def show_mask(self, mask, ax, random_color=False):
         if random_color: