Spaces:
No application file
No application file
| 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 | |
| async def predict_emotion(request: Request): | |
| prediction = model.predict([request.text])[0] | |
| return {"emotion": prediction} | |