model files are moved to hf from gdrive due to the download limit error

app.py

@@ -8,9 +8,20 @@ import utils
 from ldm.util import instantiate_from_config
 from omegaconf import OmegaConf
 from zipfile import ZipFile
-import gdown
 import os
+import requests
+import shutil
 
+def download_model(url):
+    os.makedirs("models", exist_ok=True)
+    local_filename = url.split('/')[-1]
+    with requests.get(url, stream=True) as r:
+        with open(os.path.join("models", local_filename), 'wb') as file:
+            shutil.copyfileobj(r.raw, file)
+    with ZipFile("models/gqa_inpaint.zip", 'r') as zObject:
+        zObject.extractall(path="models/")
+    os.remove("models/gqa_inpaint.zip")
+    
 MODEL = None
 
 def inference(image: np.ndarray, instruction: str, center_crop: bool):
@@ -37,11 +48,11 @@ if __name__ == "__main__":
     )
     args = parser.parse_args()
 
-    gdown.download(id="1tp0aHAS-ccrIfNz7XrGTSdNIPNZjOVSp", output="models/")
-    with ZipFile("models/gqa_inpaint.zip", 'r') as zObject:
-        zObject.extractall(path="models/")
-    os.remove("models/gqa_inpaint.zip")
-    
+    print("## Downloading the model file")
+    download_model("https://huggingface.co/abyildirim/inst-inpaint-models/resolve/main/gqa_inpaint.zip")
+    print("## Download is completed")
+
+    print("## Running the demo")
     parsed_config = OmegaConf.load(args.config)
     MODEL = instantiate_from_config(parsed_config["model"])
     model_state_dict = torch.load(args.checkpoint, map_location="cpu")["state_dict"]