fix return predictions

2023-05-27 17:18:28 +02:00 · 2023-05-27 17:18:28 +02:00 · 2f2de698b6
commit 2f2de698b6
parent cfc9b1c0d8
1 changed files with 4 additions and 0 deletions
--- a/application/services/sentiment_service.py
+++ b/application/services/sentiment_service.py
@ -13,4 +13,8 @@ def get_data():
    predicitons = sentiment_prediction(data['sentences']) #predykcje
    count_labels = count_predictions(predicitons) #dane do wykresu

+    for i in range(0, len(predicitons)):
+        predicitons[i]['sentence'] = data['sentences'][i]
+        predicitons[i]['label'] = predicitons[i]['label'][6:]
+
    return jsonify({"predictions": predicitons, "count_labels": count_labels})