Commit
·
8e14057
1
Parent(s):
f7a530f
fix: predict intent
Browse files
main.py
CHANGED
@@ -48,4 +48,4 @@ def predict_intent(request: PredictionRequest):
|
|
48 |
preds_list = text_classifier(sentence)
|
49 |
best_pred = max(preds_list[0], key=lambda x: x["score"]) # Get highest-scoring intent
|
50 |
|
51 |
-
return {"
|
|
|
48 |
preds_list = text_classifier(sentence)
|
49 |
best_pred = max(preds_list[0], key=lambda x: x["score"]) # Get highest-scoring intent
|
50 |
|
51 |
+
return {"predicted_intent": best_pred["label"], "confidence": best_pred["score"]}
|