Hugging Face's logo

Spaces:
Ehrii
/
sentiment-analysis
Running

App Files Community
Ehrii commited on
Commit
f1c88da
·
verified ·
1 Parent(s): e492073

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -66,10 +66,13 @@ def analyze_sentiment(request: SentimentRequest):
66
  label_map = {
67
  "LABEL_0": "negative",
68
  "LABEL_1": "positive",
 
69
  "0": "negative",
70
  "1": "positive",
 
71
  "NEGATIVE": "negative",
72
  "POSITIVE": "positive",
 
73
  }
74
 
75
  # Ensure label exists and map it correctly
 
66
  label_map = {
67
  "LABEL_0": "negative",
68
  "LABEL_1": "positive",
69
+ "LABEL_2": "neutral",
70
  "0": "negative",
71
  "1": "positive",
72
+ "2": "neutral",
73
  "NEGATIVE": "negative",
74
  "POSITIVE": "positive",
75
+ "NEUTRAL": "neutral",
76
  }
77
 
78
  # Ensure label exists and map it correctly