Hugging Face's logo

Spaces:
aryrk
/
SingleImageReflectionRemoval
Sleeping

App Files Community
aryrk commited on
Commit
0b76722
·
1 Parent(s): 2bb76b3

[update] update model epoch

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ os.makedirs(RESULTS_DIR, exist_ok=True)
14
  os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
15
 
16
  REPO_ID = "hasnafk/SingleImageReflectionRemoval"
17
- MODEL_FILE = "pix2pix_model.pth"
18
 
19
  model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILE, cache_dir=CHECKPOINTS_DIR)
20
 
 
14
  os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
15
 
16
  REPO_ID = "hasnafk/SingleImageReflectionRemoval"
17
+ MODEL_FILE = "310_net_G.pth"
18
 
19
  model_path = hf_hub_download(repo_id=REPO_ID, filename=MODEL_FILE, cache_dir=CHECKPOINTS_DIR)
20