src/schemas/PredictResponse.py

from typing import List
from pydantic import BaseModel

from schemas.PredictItemResult import PredictItemResult


class PredictResponse(BaseModel):
    model_name: str
    results: List[PredictItemResult]