SentimentPro_API / main.py
rabindra-sss's picture
Update main.py
dd31c7f verified
raw
history blame contribute delete
398 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from backend import predict
app = FastAPI()
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}