Spaces:
Running
Running
Josh Veitch-Michaelis
commited on
Commit
·
9a0738a
1
Parent(s):
4404cfc
update model path
Browse files
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="
|
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="
|
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="
|
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
|