Spaces:

Ysfxjo
/

arabic-complaint-classification

Runtime error

arabic-complaint-classification / interfaces /api /dependencies.py

Yousef Ammar

Add FastAPI complaint classification API

5761792 5 months ago

501 Bytes

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

	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