Spaces:

Hero0963
/

sentiment_analysis_demo_01

Runtime error

Hero0963 commited on Feb 9, 2023

Commit

f915275

•

1 Parent(s): 0798cfa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,12 +16,12 @@ class EmotionClassifier:
     def predict(self, input_text: str):
         pred = self.pipeline(input_text)[0]
         result = {
-            "Sadness 😭": pred[0]["score"],
-            "Joy 😂": pred[1]["score"],
-            "Love 😍": pred[2]["score"],
-            "Anger 😠": pred[3]["score"],
             "Fear 😨": pred[4]["score"],
-            "Surprise 😲": pred[5]["score"],
         }
         return result

     def predict(self, input_text: str):
         pred = self.pipeline(input_text)[0]
         result = {
+            "Sadness 😢": pred[0]["score"],
+            "Joy 😆": pred[1]["score"],
+            "Love 🥰": pred[2]["score"],
+            "Anger 🤬": pred[3]["score"],
             "Fear 😨": pred[4]["score"],
+            "Surprise 😯": pred[5]["score"],
         }
         return result