Zhibinhong commited on
Commit
809e94e
1 Parent(s): 5b4829a

Update visual_chatgpt.py

Browse files
Files changed (1) hide show
  1. visual_chatgpt.py +4 -2
visual_chatgpt.py CHANGED
@@ -811,8 +811,10 @@ class Segmenting:
811
  print("find",os.system("find ./ -name 'handler.py'"))
812
  print("finddir",os.system("find ./ -type d -iname 'checkpoints'"))
813
  url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
814
- if not os.path.exists(self.model_checkpoint_path):
815
- wget.download(url,out="repository/"+self.model_checkpoint_path)
 
 
816
 
817
  def show_mask(self, mask, ax, random_color=False):
818
  if random_color:
 
811
  print("find",os.system("find ./ -name 'handler.py'"))
812
  print("finddir",os.system("find ./ -type d -iname 'checkpoints'"))
813
  url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
814
+ if not os.path.exists("/repository/checkpoints"):
815
+ os.makedirs("/repository/checkpoints")
816
+ # wget.download(url,out=self.model_checkpoint_path)
817
+ wget.download(url,out="/repository/checkpoints/sam")
818
 
819
  def show_mask(self, mask, ax, random_color=False):
820
  if random_color: