Alejadro Sanchez-Giraldo commited on
Commit
77d5f1f
β€’
1 Parent(s): db12616

added new options base on new LLM

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -27,7 +27,12 @@ def translate_label(label):
27
  label_mapping = {
28
  "LABEL_0": "🀬 Negative",
29
  "LABEL_1": "😢 Neutral",
30
- "LABEL_2": "πŸ˜ƒ Positive"
 
 
 
 
 
31
  }
32
  return label_mapping.get(label, "Unknown")
33
 
 
27
  label_mapping = {
28
  "LABEL_0": "🀬 Negative",
29
  "LABEL_1": "😢 Neutral",
30
+ "LABEL_2": "πŸ˜ƒ Positive",
31
+ "1 star": "🀬 Negative",
32
+ "2 stars": "🀬 Negative",
33
+ "3 stars": "😢 Neutral",
34
+ "4 stars": "πŸ˜ƒ Positive",
35
+ "5 stars": "πŸ˜ƒ Positive"
36
  }
37
  return label_mapping.get(label, "Unknown")
38