Spaces:
Running
Running
Commit Β·
8559887
1
Parent(s): abd0b64
Set default model paths to load actual ML models instead of demo mode
Browse files- backend/__init__.py +2 -2
backend/__init__.py
CHANGED
|
@@ -84,8 +84,8 @@ def create_app(config_override: dict | None = None) -> Flask:
|
|
| 84 |
# ββ ML model loading βββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 85 |
from .ml.ml_engine import init_models
|
| 86 |
init_models(
|
| 87 |
-
psycho_model_dir=os.environ.get("PSYCHO_MODEL_DIR", ""),
|
| 88 |
-
roberta_ckpt =os.environ.get("ROBERTA_CKPT", ""),
|
| 89 |
)
|
| 90 |
|
| 91 |
# ββ Background scheduler βββββββββββββββββββββββββββββββββββββββββββββββββ
|
|
|
|
| 84 |
# ββ ML model loading βββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 85 |
from .ml.ml_engine import init_models
|
| 86 |
init_models(
|
| 87 |
+
psycho_model_dir=os.environ.get("PSYCHO_MODEL_DIR", "models/psychometric"),
|
| 88 |
+
roberta_ckpt =os.environ.get("ROBERTA_CKPT", "models/text/roberta-model.pt"),
|
| 89 |
)
|
| 90 |
|
| 91 |
# ββ Background scheduler βββββββββββββββββββββββββββββββββββββββββββββββββ
|