Spaces:
Running
Running
Update mic_worker.py
Browse files- 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(
|
| 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 |
|