DeepakKumarMSL commited on
Commit
9c3068d
·
verified ·
1 Parent(s): 39c2d5b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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., joy, anger, sadness) **without training**, by framing it as a textual entailment task.
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 = ["joy", "anger", "sadness", "fear", "surprise", "neutral"]
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