Hugging Face's logo

Spaces:
markmagic
/
SeeMoreDetails
Running on Zero

App Files
Eduard-Sebastian Zamfir commited on
Commit
dae1a1c
•
1 Parent(s): 9080570

load checkpoint from huggingface

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +2 -1
.gitignore CHANGED
@@ -1,2 +1,3 @@
1
  __pycache__/
2
  flagged/
 
 
1
  __pycache__/
2
  flagged/
3
+ checkpoints
app.py CHANGED
@@ -65,7 +65,8 @@ def load_network(net, load_path, strict=True, param_key='params'):
65
  net.load_state_dict(load_net, strict=strict)
66
 
67
  CONFIG = "configs/eval_seemore_t_x4.yml"
68
- MODEL_NAME = "checkpoints/SeemoRe_T/X4/net_g_latest.pth"
 
69
 
70
  # parse config file
71
  with open(os.path.join(CONFIG), "r") as f:
 
65
  net.load_state_dict(load_net, strict=strict)
66
 
67
  CONFIG = "configs/eval_seemore_t_x4.yml"
68
+ hf_hub_download(repo_id="eduardzamfir/SeemoRe-T", filename="SeemoRe_T_X4.pth", local_dir="./")
69
+ MODEL_NAME = "SeemoRe_T_X4.pth"
70
 
71
  # parse config file
72
  with open(os.path.join(CONFIG), "r") as f: