Upload folder using huggingface_hub
Browse files- app/api/routes.py +7 -0
app/api/routes.py
CHANGED
|
@@ -353,11 +353,18 @@ async def ml_status():
|
|
| 353 |
from ..ml.runtime import get_model
|
| 354 |
|
| 355 |
model = get_model()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 356 |
return {
|
| 357 |
"enabled": settings.ml_disposable_enabled,
|
| 358 |
"repo": settings.ml_model_repo,
|
| 359 |
"loaded": bool(model is not None and model.loaded),
|
| 360 |
"last_error": getattr(model, "last_error", None) if model else None,
|
|
|
|
| 361 |
}
|
| 362 |
|
| 363 |
|
|
|
|
| 353 |
from ..ml.runtime import get_model
|
| 354 |
|
| 355 |
model = get_model()
|
| 356 |
+
sample = None
|
| 357 |
+
if model is not None and model.loaded:
|
| 358 |
+
try:
|
| 359 |
+
sample = model.predict(["john.smith [SEP] stripe.com"])
|
| 360 |
+
except Exception as exc: # pragma: no cover - defensive
|
| 361 |
+
sample = [f"predict raised: {exc}"]
|
| 362 |
return {
|
| 363 |
"enabled": settings.ml_disposable_enabled,
|
| 364 |
"repo": settings.ml_model_repo,
|
| 365 |
"loaded": bool(model is not None and model.loaded),
|
| 366 |
"last_error": getattr(model, "last_error", None) if model else None,
|
| 367 |
+
"predict_sample": sample,
|
| 368 |
}
|
| 369 |
|
| 370 |
|