Update README.md
Browse files
README.md
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
|
| 3 |
This project demonstrates how to perform **Tone Detection** using the [`facebook/bart-large-mnli`](https://huggingface.co/facebook/bart-large-mnli) model through **zero-shot classification** based on Natural Language Inference (NLI).
|
| 4 |
|
| 5 |
-
This approach enables you to classify emotional tone (e.g.,
|
| 6 |
|
| 7 |
---
|
| 8 |
|
|
@@ -31,7 +31,7 @@ from transformers import pipeline
|
|
| 31 |
|
| 32 |
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
| 33 |
|
| 34 |
-
labels = ["
|
| 35 |
|
| 36 |
text = "I can't believe this is happening again. So frustrating."
|
| 37 |
|
|
|
|
| 2 |
|
| 3 |
This project demonstrates how to perform **Tone Detection** using the [`facebook/bart-large-mnli`](https://huggingface.co/facebook/bart-large-mnli) model through **zero-shot classification** based on Natural Language Inference (NLI).
|
| 4 |
|
| 5 |
+
This approach enables you to classify emotional tone (e.g., Positive, Negative, Neutral, etc.) **without training**, by framing it as a textual entailment task.
|
| 6 |
|
| 7 |
---
|
| 8 |
|
|
|
|
| 31 |
|
| 32 |
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
|
| 33 |
|
| 34 |
+
labels = ["positive", "neutral", "negative"]
|
| 35 |
|
| 36 |
text = "I can't believe this is happening again. So frustrating."
|
| 37 |
|