Muteeba commited on
Commit
66291d9
Β·
1 Parent(s): 3383b36

fix: correct paths /app/data/ for git files, /tmp/ for downloads

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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","/data/esm2_cache")
10
- os.environ.setdefault("FUNGO_EMB_CACHE", "/data/embedding_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 = "/data/models"
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"]