pan-card-classifier / api /schemas.py
shailgsits's picture
Upload folder using huggingface_hub
f0e10d7 verified
Raw
History Blame Contribute Delete
239 Bytes
from pydantic import BaseModel
class PredictionResult(BaseModel):
label: str
confidence: float
class PredictionResponse(BaseModel):
success: bool
prediction: PredictionResult
model: str
processing_time_ms: float