aman-s-affinsys commited on
Commit
8e14057
·
1 Parent(s): f7a530f

fix: predict intent

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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 {"intent": best_pred["label"], "confidence": best_pred["score"]}
 
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"]}