expon-emotions / src /api_emotions.py
alexander1010's picture
Add src folder
d7aa146 verified
Raw
History Blame Contribute Delete
418 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from src.infer import predict
app = FastAPI(title="Expon Emotions Service")
class InputText(BaseModel):
transcript: str
@app.post("/analysis/emotions")
def analyze_emotions(item: InputText):
label, prob, dist = predict(item.transcript)
return {
"dominant_emotion": label,
"probability": prob,
"distribution": dist
}