Spaces:
Sleeping
Sleeping
File size: 465 Bytes
1e5b98a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# api/model_loader.py
from functools import lru_cache
import joblib
from .config import MODEL_PATH
class ModelNotFound(Exception):
pass
@lru_cache(maxsize=1)
def get_model():
"""
Carga el modelo entrenado desde disco y lo cachea.
"""
if not MODEL_PATH.exists():
raise ModelNotFound(f"No se encontró el modelo en {MODEL_PATH}. Ejecuta src/train.py primero.")
model = joblib.load(MODEL_PATH)
return model
|