eliphatfs commited on
Commit
796d131
1 Parent(s): 3f3b760
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -30,6 +30,7 @@ class SAMAPI:
30
  if sam_checkpoint is None:
31
  sam_checkpoint = "tmp/sam_vit_h_4b8939.pth"
32
  if not os.path.exists(sam_checkpoint):
 
33
  urllib.request.urlretrieve(
34
  "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
35
  sam_checkpoint
 
30
  if sam_checkpoint is None:
31
  sam_checkpoint = "tmp/sam_vit_h_4b8939.pth"
32
  if not os.path.exists(sam_checkpoint):
33
+ os.makedirs('tmp', exist_ok=True)
34
  urllib.request.urlretrieve(
35
  "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
36
  sam_checkpoint