Josh Veitch-Michaelis commited on
Commit
9a0738a
·
1 Parent(s): 4404cfc

update model path

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,21 +21,21 @@ if __name__ == '__main__':
21
 
22
  if not args.model_path:
23
  model_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
24
- filename="ckpt_epoch_35.pth",
25
  token=os.environ["HUGGINGFACE_TOKEN"])
26
  else:
27
  model_path = args.model_path
28
 
29
  if not args.cfg:
30
  model_config = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
31
- filename="MetaFG_2_224.yaml",
32
  token=os.environ["HUGGINGFACE_TOKEN"])
33
  else:
34
  model_config = args.cfg
35
 
36
  if not args.names_path:
37
  names_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
38
- filename="names_mf2.txt",
39
  token=os.environ["HUGGINGFACE_TOKEN"])
40
  else:
41
  names_path = args.names_path
 
21
 
22
  if not args.model_path:
23
  model_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
24
+ filename="inat_sgd_6k.pth",
25
  token=os.environ["HUGGINGFACE_TOKEN"])
26
  else:
27
  model_path = args.model_path
28
 
29
  if not args.cfg:
30
  model_config = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
31
+ filename="MetaFG_2_384_inat.yaml",
32
  token=os.environ["HUGGINGFACE_TOKEN"])
33
  else:
34
  model_config = args.cfg
35
 
36
  if not args.names_path:
37
  names_path = hf_hub_download(repo_id="joshvm/inaturalist_sgd_4k",
38
+ filename="inat_sgd_names.txt",
39
  token=os.environ["HUGGINGFACE_TOKEN"])
40
  else:
41
  names_path = args.names_path