Commit
•
12d8e9f
1
Parent(s):
212790e
Update visual_chatgpt.py
Browse files- visual_chatgpt.py +2 -0
visual_chatgpt.py
CHANGED
|
@@ -805,6 +805,8 @@ class Segmenting:
|
|
| 805 |
self.mask_generator = SamAutomaticMaskGenerator(self.sam)
|
| 806 |
|
| 807 |
def download_parameters(self):
|
|
|
|
|
|
|
| 808 |
url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
|
| 809 |
if not os.path.exists(self.model_checkpoint_path):
|
| 810 |
wget.download(url,out="repository/"+self.model_checkpoint_path)
|
|
|
|
| 805 |
self.mask_generator = SamAutomaticMaskGenerator(self.sam)
|
| 806 |
|
| 807 |
def download_parameters(self):
|
| 808 |
+
##############
|
| 809 |
+
print("getcwd",os.getcwd())
|
| 810 |
url = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
|
| 811 |
if not os.path.exists(self.model_checkpoint_path):
|
| 812 |
wget.download(url,out="repository/"+self.model_checkpoint_path)
|