src/schemas/PredictRequest.py

from typing import List
from pydantic import BaseModel

from src.schemas.ModelFeatures import ModelFeatures

class PredictRequest(BaseModel):
    model_name: str
    inputs: List[ModelFeatures]