Spaces:

nkigumnov
/

banks-ethics-sentiment

Build error

nkigumnov commited on Apr 18, 2023

Commit

703d00e

1 Parent(s): 2a032a9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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]),
     }
     return (
         df["1"],

         "1": float(prob[1][0][2]),
         "0": float(prob[1][0][1]),
         "-1": float(prob[1][0][0]),
+        "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"],