pixano.app.api.models

get_models(settings) async

Load models

Returns:

Type	Description
list[str]	List of models

Source code in pixano/app/api/models.py

@router.get("/models", response_model=list[str])
async def get_models(
    settings: Annotated[Settings, Depends(get_settings)],
) -> list[str]:
    """Load models

    Returns:
        list[str]: List of models
    """

    # Load list of models
    models = []
    for model_path in settings.model_dir.glob("*.onnx"):
        models.append(model_path.name)

    # Return list of models
    if models:
        return models
    raise HTTPException(
        status_code=404,
        detail=f"No models found in {settings.model_dir.absolute()}",
    )