File size: 196 Bytes
6ab520d
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
from pydantic import BaseModel


class PredictRequest(BaseModel):
    file: str


class PredictResponse(BaseModel):
    success: bool
    prediction: str
    score: float
    image_file_name: str