emotion_api / main.py
ikshit2004's picture
Create main.py
0fbe057 verified
from fastapi import FastAPI
from pydantic import BaseModel
import joblib
app = FastAPI()
# Load your model at startup
model = joblib.load("app/emotion_classifier_pipe_lr.pkl")
class Request(BaseModel):
text: str
@app.post("/predict")
async def predict_emotion(request: Request):
prediction = model.predict([request.text])[0]
return {"emotion": prediction}