fix: PKL_DIR points to /tmp/models where files download
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
import csv, io, logging, os, re as _re, sys, time
|
| 3 |
from collections import OrderedDict
|
| 4 |
|
| 5 |
-
os.environ.setdefault("FUNGO_PKL_DIR", "/
|
| 6 |
os.environ.setdefault("FUNGO_VOCAB_PKL", "/app/data/labels/vocabularies.pkl")
|
| 7 |
os.environ.setdefault("FUNGO_IA_PKL", "/app/data/go_data/ia_weights.pkl")
|
| 8 |
os.environ.setdefault("FUNGO_FEAT_META", "/app/data/features/feature_metadata.json")
|
|
@@ -57,6 +57,7 @@ def download_models_if_needed():
|
|
| 57 |
continue
|
| 58 |
log.info(" Downloading %s ...", fname)
|
| 59 |
hf_hub_download(
|
|
|
|
| 60 |
repo_id="Muteeba/FunGO-models",
|
| 61 |
filename=fname,
|
| 62 |
repo_type="model",
|
|
|
|
| 2 |
import csv, io, logging, os, re as _re, sys, time
|
| 3 |
from collections import OrderedDict
|
| 4 |
|
| 5 |
+
os.environ.setdefault("FUNGO_PKL_DIR", "/tmp/models")
|
| 6 |
os.environ.setdefault("FUNGO_VOCAB_PKL", "/app/data/labels/vocabularies.pkl")
|
| 7 |
os.environ.setdefault("FUNGO_IA_PKL", "/app/data/go_data/ia_weights.pkl")
|
| 8 |
os.environ.setdefault("FUNGO_FEAT_META", "/app/data/features/feature_metadata.json")
|
|
|
|
| 57 |
continue
|
| 58 |
log.info(" Downloading %s ...", fname)
|
| 59 |
hf_hub_download(
|
| 60 |
+
token=os.environ.get("HF_TOKEN"),
|
| 61 |
repo_id="Muteeba/FunGO-models",
|
| 62 |
filename=fname,
|
| 63 |
repo_type="model",
|