Arvind111 commited on
Commit
1158562
ยท
verified ยท
1 Parent(s): 7f00f1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,9 +27,9 @@ def predict_sentiment(text):
27
  # --- NEW CODE: Map prediction to labels ---
28
  # 3. Define the mapping from numbers to labels
29
  sentiment_map = {
30
- 0: "Neutral ๐Ÿ˜",
31
- 1: "Happy ๐Ÿ˜Š",
32
- 2: "Sad ๐Ÿ˜ž"
33
  }
34
 
35
  # 4. Get the label from the map.
 
27
  # --- NEW CODE: Map prediction to labels ---
28
  # 3. Define the mapping from numbers to labels
29
  sentiment_map = {
30
+ 0: "NEUTRAL",
31
+ 1: "HAPPY",
32
+ 2: "SAD"
33
  }
34
 
35
  # 4. Get the label from the map.