TextRephrase / models /rephraseparam.py
SoumyaJ's picture
Changes in rephrase
0da1f7c
raw
history blame contribute delete
502 Bytes
from pydantic import BaseModel, Field
from typing import Optional
class rephraseparam(BaseModel):
texttorephrase: str = Field(..., description="input text for summarization")
tone : Optional[str] = Field('professional', description="text summary will follow tone: casual/professional/straightforward")
maxtokens : Optional[int] = Field(200, description="max length for the output")
language : Optional[str] = Field("english", description="summary needs to be in any specific language")