Fix issue with app.py

backend/app.py

@@ -6,10 +6,6 @@ from fastapi.responses import FileResponse
 from transformers import GPT2Tokenizer, GPT2Model, pipeline
 import torch as t
 import logging
-import pydantic
-from pydantic import BaseModel
-class TextRequest(BaseModel):
-    text: str
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -38,15 +34,13 @@ except Exception as e:
     raise
 
 @app.post("/process")
-async def process_text(payload: TextRequest):
+async def process_text(text: str = Body(..., embed=True)):
     """
     Process the input text:
       - Tokenizes the text
       - Runs the GPT2 model to obtain attentions
       - Returns the tokens and attention values (rounded to 2 decimals)
     """
-
-    text = payload.txt
     try:
         logger.info(f"Received text: {text}")
         # Tokenize input text (truncating if needed)