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
|