Update main.py

main.py

@@ -1,19 +1,18 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-from backend import predict
-from mangum import Mangum
-
-app = FastAPI()
-
-handler = Mangum(app)
-class Input(BaseModel):
-    text: str
-
-@app.get("/")
-async def root():
-    return {"message": "Sentiment Analysis API is running!"}
-
-@app.post("/predict/")
-async def predict_sentiment(input: Input):
-    prediction = predict(input.text)
-    return {"text": input.text, "sentiment": prediction}
+from fastapi import FastAPI
+from pydantic import BaseModel
+from backend import predict
+
+app = FastAPI()
+
+handler = Mangum(app)
+class Input(BaseModel):
+    text: str
+
+@app.get("/")
+async def root():
+    return {"message": "Sentiment Analysis API is running!"}
+
+@app.post("/predict/")
+async def predict_sentiment(input: Input):
+    prediction = predict(input.text)
+    return {"text": input.text, "sentiment": prediction}