Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
c704d5b
1
Parent(s):
2072beb
Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ class SAMAPI:
|
|
72 |
if sam_checkpoint is None:
|
73 |
sam_checkpoint = "sam_pt/sam_vit_h_4b8939.pth"
|
74 |
if not os.path.exists(sam_checkpoint):
|
75 |
-
os.makedirs('
|
76 |
urllib.request.urlretrieve(
|
77 |
"https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
|
78 |
sam_checkpoint
|
|
|
72 |
if sam_checkpoint is None:
|
73 |
sam_checkpoint = "sam_pt/sam_vit_h_4b8939.pth"
|
74 |
if not os.path.exists(sam_checkpoint):
|
75 |
+
os.makedirs('sam_pt', exist_ok=True)
|
76 |
urllib.request.urlretrieve(
|
77 |
"https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
|
78 |
sam_checkpoint
|