fix: correct paths /app/data/ for git files, /tmp/ for downloads
Browse files
app.py
CHANGED
|
@@ -2,12 +2,12 @@
|
|
| 2 |
import csv, io, logging, os, re as _re, sys, time
|
| 3 |
from collections import OrderedDict
|
| 4 |
|
| 5 |
-
os.environ.setdefault("FUNGO_PKL_DIR", "/data/models")
|
| 6 |
-
os.environ.setdefault("FUNGO_VOCAB_PKL", "/data/labels/vocabularies.pkl")
|
| 7 |
-
os.environ.setdefault("FUNGO_IA_PKL", "/data/go_data/ia_weights.pkl")
|
| 8 |
-
os.environ.setdefault("FUNGO_FEAT_META", "/data/features/feature_metadata.json")
|
| 9 |
-
os.environ.setdefault("FUNGO_MODEL_CACHE","/
|
| 10 |
-
os.environ.setdefault("FUNGO_EMB_CACHE", "/
|
| 11 |
os.environ.setdefault("FUNGO_OFFLINE", "0")
|
| 12 |
os.environ.setdefault("FUNGO_DEBUG", "0")
|
| 13 |
os.environ.setdefault("FUNGO_PORT", "7860")
|
|
@@ -35,7 +35,7 @@ _models_ready = False
|
|
| 35 |
# ββ Download models from HF Model Repo βββββββββββββββββββββββ
|
| 36 |
def download_models_if_needed():
|
| 37 |
"""Download pkl files from Muteeba/FunGO-models if not present."""
|
| 38 |
-
models_dir = "/
|
| 39 |
os.makedirs(models_dir, exist_ok=True)
|
| 40 |
|
| 41 |
files_needed = ["models_BPO.pkl", "models_MFO.pkl", "models_CCO.pkl"]
|
|
|
|
| 2 |
import csv, io, logging, os, re as _re, sys, time
|
| 3 |
from collections import OrderedDict
|
| 4 |
|
| 5 |
+
os.environ.setdefault("FUNGO_PKL_DIR", "/app/data/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")
|
| 9 |
+
os.environ.setdefault("FUNGO_MODEL_CACHE","/tmp/esm2_cache")
|
| 10 |
+
os.environ.setdefault("FUNGO_EMB_CACHE", "/tmp/embedding_cache")
|
| 11 |
os.environ.setdefault("FUNGO_OFFLINE", "0")
|
| 12 |
os.environ.setdefault("FUNGO_DEBUG", "0")
|
| 13 |
os.environ.setdefault("FUNGO_PORT", "7860")
|
|
|
|
| 35 |
# ββ Download models from HF Model Repo βββββββββββββββββββββββ
|
| 36 |
def download_models_if_needed():
|
| 37 |
"""Download pkl files from Muteeba/FunGO-models if not present."""
|
| 38 |
+
models_dir = "/tmp/models"
|
| 39 |
os.makedirs(models_dir, exist_ok=True)
|
| 40 |
|
| 41 |
files_needed = ["models_BPO.pkl", "models_MFO.pkl", "models_CCO.pkl"]
|