doevent commited on
Commit
afd2efd
1 Parent(s): 51f6f9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -25,8 +25,9 @@ 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
+ # 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
+ os.system("mv isnet.pth saved_models/")
31
 
32
  class GOSNormalize(object):
33
  '''