File size: 325 Bytes
625daea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
"""Pydantic models for text generation"""
from pydantic import BaseModel
from typing import Optional
class GenerateRequest(BaseModel):
input_text: str
max_length: Optional[int] = 128
num_beams: Optional[int] = 4
class GenerateResponse(BaseModel):
input_text: str
generated_text: str
timestamp: str
|