From 97cb15e51d67bd8e809ec1117a8b1dadcadbfeff Mon Sep 17 00:00:00 2001
From: Max Skobeev <doevent@users.noreply.huggingface.co>
Date: Mon, 30 Oct 2023 15:56:15 +0000
Subject: [PATCH] fix official weights

The model is no longer available from the link
---
 app.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

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):
     '''