Update app.py
Browse files
app.py
CHANGED
|
@@ -31,9 +31,9 @@ tokenizer = BertTokenizer.from_pretrained("shrish191/sentiment-bert")
|
|
| 31 |
|
| 32 |
# Manually define the correct mapping
|
| 33 |
LABELS = {
|
| 34 |
-
0: "
|
| 35 |
-
1: "
|
| 36 |
-
2: "
|
| 37 |
}
|
| 38 |
|
| 39 |
def classify_sentiment(text):
|
|
|
|
| 31 |
|
| 32 |
# Manually define the correct mapping
|
| 33 |
LABELS = {
|
| 34 |
+
0: "Neutral",
|
| 35 |
+
1: "Positive",
|
| 36 |
+
2: "Negative"
|
| 37 |
}
|
| 38 |
|
| 39 |
def classify_sentiment(text):
|