Update README.md
Browse files
README.md
CHANGED
|
@@ -42,15 +42,15 @@ tokenizer = AutoTokenizer.from_pretrained("fine_tuned_finsetiment_model")
|
|
| 42 |
|
| 43 |
# Define the label mapping as provided
|
| 44 |
label_mapping_reverse = {
|
| 45 |
-
'
|
| 46 |
-
'
|
| 47 |
-
'
|
| 48 |
-
'
|
| 49 |
-
'
|
| 50 |
-
'
|
| 51 |
-
'
|
| 52 |
-
'
|
| 53 |
-
'
|
| 54 |
}
|
| 55 |
|
| 56 |
def model_predict(text):
|
|
|
|
| 42 |
|
| 43 |
# Define the label mapping as provided
|
| 44 |
label_mapping_reverse = {
|
| 45 |
+
'0': 'Mildly positive',
|
| 46 |
+
'1': 'Mildly negative',
|
| 47 |
+
'2': 'Moderately negative',
|
| 48 |
+
'3': 'Moderately positive',
|
| 49 |
+
'4': 'Positive',
|
| 50 |
+
'5': 'Negative',
|
| 51 |
+
'6': 'Neutral',
|
| 52 |
+
'7': 'Strongly negative',
|
| 53 |
+
'8': 'Strongly positive'
|
| 54 |
}
|
| 55 |
|
| 56 |
def model_predict(text):
|