nonzeroexit commited on
Commit
87f0d31
·
verified ·
1 Parent(s): bd01e5d

Update mic_worker.py

Browse files
Files changed (1) hide show
  1. mic_worker.py +4 -1
mic_worker.py CHANGED
@@ -19,6 +19,9 @@ import os
19
  os.environ.setdefault("OMP_NUM_THREADS", "1")
20
  os.environ.setdefault("MKL_NUM_THREADS", "1")
21
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
 
 
22
 
23
  import sys
24
  import json
@@ -75,7 +78,7 @@ def main():
75
  transformed = joblib.load(cfg["pca"]).transform(scaled) if cfg["pca"] else scaled
76
  mic_model = joblib.load(cfg["model"])
77
  mic_log = mic_model.predict(transformed)[0]
78
- mic_results[bacterium] = round(max(0.0, expm1(float(mic_log))), 3)
79
  except Exception as e:
80
  mic_results[bacterium] = f"Error: {str(e)}"
81
 
 
19
  os.environ.setdefault("OMP_NUM_THREADS", "1")
20
  os.environ.setdefault("MKL_NUM_THREADS", "1")
21
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
22
+ # Cache ProtBert in a stable location so it is downloaded only ONCE and reused
23
+ # on every subsequent prediction (instead of re-downloading each run).
24
+ os.environ.setdefault("HF_HOME", "/app/.cache/huggingface")
25
 
26
  import sys
27
  import json
 
78
  transformed = joblib.load(cfg["pca"]).transform(scaled) if cfg["pca"] else scaled
79
  mic_model = joblib.load(cfg["model"])
80
  mic_log = mic_model.predict(transformed)[0]
81
+ mic_results[bacterium] = round(expm1(float(mic_log)), 3)
82
  except Exception as e:
83
  mic_results[bacterium] = f"Error: {str(e)}"
84