Update main.py

main.py

@@ -6,6 +6,9 @@ from transformers import pipeline
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import RedirectResponse
 
+# Data model
+class TextInput(BaseModel):
+    text: str
 
 # Initialize FastAPI app and download necessary NLTK resources
 app = FastAPI()
@@ -35,6 +38,3 @@ async def predict_sentiment(text: TextInput):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# Data model
-class TextInput(BaseModel):
-    text: str