Spaces:
Runtime error
Runtime error
File size: 501 Bytes
5761792 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from fastapi import HTTPException, Request
def get_model_loader(request: Request):
"""Returns the loaded ModelLoader or 503 if models failed to load (e.g. missing HF_TOKEN)."""
loader = getattr(request.app.state, "model_loader", None)
if loader is None or not loader.is_ready():
raise HTTPException(status_code=503, detail={
"error_code": "MODELS_NOT_READY",
"message": "Models are not loaded or failed to load"
})
else:
return loader
|