AfroLogicInsect commited on
Commit
9d9e6ed
Β·
verified Β·
1 Parent(s): c9e1d41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -10,7 +10,16 @@ def predict_sentiment(text):
10
  return "Please enter some text", 0.0
11
 
12
  result = sentiment_pipeline(text)[0]
13
- label = "😊 Positive" if result['label'] == 'LABEL_1' else "😞 Negative"
 
 
 
 
 
 
 
 
 
14
  return label, round(result['score'], 3)
15
 
16
  iface = gr.Interface(
 
10
  return "Please enter some text", 0.0
11
 
12
  result = sentiment_pipeline(text)[0]
13
+
14
+ # Fixed label interpretation
15
+ if result['label'] == 'POSITIVE':
16
+ label = "😊 Positive"
17
+ elif result['label'] == 'NEGATIVE':
18
+ label = "😞 Negative"
19
+ else:
20
+ # Fallback for LABEL_0/LABEL_1
21
+ label = "😊 Positive" if result['label'] == 'LABEL_1' else "😞 Negative"
22
+
23
  return label, round(result['score'], 3)
24
 
25
  iface = gr.Interface(