hf hub added

app.py

@@ -1,24 +1,20 @@
-import requests
-import os
 import gradio as gr
 import numpy as np
 import torch
 import torchvision.models as models
+from huggingface_hub import hf_hub_url, cached_download
 
 from configs.default import get_cfg_defaults
 from modeling.build import build_model
 from utils.data_utils import linear_scaling
 
 
-url = "https://www.dropbox.com/s/uxvax5sjx5iysyl/cifr.pth?dl=0"
-r = requests.get(url, stream=True)
-if not os.path.exists("cifr.pth"):
-    with open("cifr.pth", 'wb') as f:
-        for data in r:
-            f.write(data)
+url = hf_hub_url(repo_id="birdortyedi/cifr", filename="cifr.pth")
+model_path = cached_download(url)
+
 
 cfg = get_cfg_defaults()
-cfg.MODEL.CKPT = "cifr.pth"
+cfg.MODEL.CKPT = model_path
 net, _ = build_model(cfg)
 net = net.eval()
 vgg16 = models.vgg16(pretrained=True).features.eval()