Update app.py

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI
-from pydantic import BaseModel
+from pydantic import BaseModel # для автоматического разбора
 from transformers import pipeline
 import os
 import asyncio
@@ -13,11 +13,12 @@ async def lifespan(app: FastAPI):
 
 app = FastAPI(lifespan=lifespan)
 
-class TextRequest(BaseModel):
+class Request(BaseModel):
     text: str
+    number: int
 
 @app.post("/message")
-async def model(request: TextRequest):
+async def model(request: Request):
     try:
         # Возвращаем текст в верхнем регистре
         return {"result": request.text.upper()}