Embedded / schema /schemas.py
yamunasivan's picture
create a new file
d1e319b
raw
history blame contribute delete
136 Bytes
from pydantic import BaseModel
class PredictionInput(BaseModel):
text: str
class PredictionOutput(BaseModel):
prediction: str