Muteeba commited on
Commit
b417c18
·
1 Parent(s): 66291d9

fix: PKL_DIR points to /tmp/models where files download

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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", "/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")
@@ -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",