Zhibinhong
commited on
Commit
•
809e94e
1
Parent(s):
5b4829a
Update visual_chatgpt.py
Browse files- 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(
|
815 |
-
|
|
|
|
|
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:
|