| """Prediction API routes.""" |
|
|
| from fastapi import APIRouter, Depends |
|
|
| from app.api.controllers import PredictionController |
| from app.api.models import ImageRequest, PredictionResponse |
| from app.core.dependencies import get_inference_service |
| from app.services.base import InferenceService |
|
|
| router = APIRouter() |
|
|
|
|
| @router.post("/predict", response_model=PredictionResponse) |
| async def predict( |
| request: ImageRequest, |
| service: InferenceService = Depends(get_inference_service) |
| ): |
| """ |
| Run inference on base64-encoded image. |
| |
| Returns prediction, confidence, predicted label, model name, and media type. |
| """ |
| return await PredictionController.predict(request, service) |
|
|
