sitcod3 commited on
Commit
cf368ae
·
verified ·
1 Parent(s): 9a67bad

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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