Update app.py

app.py

@@ -25,7 +25,8 @@ 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"
+    # MODEL_PATH_URL = "https://drive.google.com/uc?id=1KyMpRjewZdyYfxHPYcd-ZbanIXtin0Sn" # IS-Net
+    MODEL_PATH_URL = "https://drive.google.com/uc?id=1nV57qKuy--d5u1yvkng9aXW1KS4sOpOi" # IS-Net General Use
     gdown.download(MODEL_PATH_URL, "saved_models/isnet.pth", use_cookies=False)
     
 class GOSNormalize(object):