diff --git a/app.py b/app.py index 6024d1a..4f764bc 100644 --- a/app.py +++ b/app.py @@ -25,8 +25,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu' # Download official weights if not os.path.exists("saved_models"): os.mkdir("saved_models") - MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn" - gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False) + os.system("mv isnet.pth saved_models/") class GOSNormalize(object): '''