Update app.py

app.py

@@ -1,10 +1,14 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 import torch
+from pydantic import BaseModel
 
 app = FastAPI()
 
 
+class InputText(BaseModel):
+    text : str
+
 
 model_name = "cardiffnlp/twitter-xlm-roberta-base-sentiment"
 sentiment_model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -17,7 +21,8 @@ def greet_json():
 
 
 @app.post("/sentiment_score")
-async def sentiment_score(text: str):
+async def sentiment_score(inp: InputText):
+    text = inp.text
     inputs = sentiment_tokenizer(text[:2500], return_tensors='pt')
     
     with torch.no_grad():