flamehaze1115 commited on
Commit
c704d5b
1 Parent(s): 2072beb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('tmp', exist_ok=True)
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