load checkpoint from huggingface

.gitignore

@@ -1,2 +1,3 @@
 __pycache__/
 flagged/
+checkpoints

app.py

@@ -65,7 +65,8 @@ def load_network(net, load_path, strict=True, param_key='params'):
     net.load_state_dict(load_net, strict=strict)
 
 CONFIG     = "configs/eval_seemore_t_x4.yml"
-MODEL_NAME = "checkpoints/SeemoRe_T/X4/net_g_latest.pth"
+hf_hub_download(repo_id="eduardzamfir/SeemoRe-T", filename="SeemoRe_T_X4.pth", local_dir="./")
+MODEL_NAME = "SeemoRe_T_X4.pth"
 
 # parse config file
 with open(os.path.join(CONFIG), "r") as f: