doevent commited on
Commit
97cb15e
1 Parent(s): 5e6d9e0

fix official weights

Browse files

The model is no longer available from the link

Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -25,8 +25,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
25
  # Download official weights
26
  if not os.path.exists("saved_models"):
27
  os.mkdir("saved_models")
28
- MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn"
29
- gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False)
30
 
31
  class GOSNormalize(object):
32
  '''
 
25
  # Download official weights
26
  if not os.path.exists("saved_models"):
27
  os.mkdir("saved_models")
28
+ os.system("mv isnet.pth saved_models/")
 
29
 
30
  class GOSNormalize(object):
31
  '''