Update app.py

app.py

@@ -10,10 +10,10 @@ def predict(sentence: str):
         "1": float(prob[1][0][2]),
         "0": float(prob[1][0][1]),
         "-1": float(prob[1][0][0]),
-        "Communication": float(prob[0][0][0]),
-        "Quality": float(prob[0][0][1]),
-        "Price": float(prob[0][0][2]),
-        "Safety": float(prob[0][0][3]),
+        "Communication": float(prob[0][0][1]),
+        "Quality": float(prob[0][0][2]),
+        "Price": float(prob[0][0][3]),
+        "Safety": float(prob[0][0][4]),
     }
     return (
         df["1"],