Update README.md
Browse files
README.md
CHANGED
@@ -49,6 +49,6 @@ It was tested on SENTIPOLC 2016 test set
|
|
49 |
```
|
50 |
model = AutoModelForSequenceClassification.from_pretrained('aequa-tech/sentiment-it',num_labels=3, ignore_mismatched_sizes=True)
|
51 |
tokenizer = AutoTokenizer.from_pretrained("m-polignano-uniba/bert_uncased_L-12_H-768_A-12_italian_alb3rt0")
|
52 |
-
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
53 |
classifier("L'insostenibile leggerezza dell'essere")
|
54 |
```
|
|
|
49 |
```
|
50 |
model = AutoModelForSequenceClassification.from_pretrained('aequa-tech/sentiment-it',num_labels=3, ignore_mismatched_sizes=True)
|
51 |
tokenizer = AutoTokenizer.from_pretrained("m-polignano-uniba/bert_uncased_L-12_H-768_A-12_italian_alb3rt0")
|
52 |
+
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=None)
|
53 |
classifier("L'insostenibile leggerezza dell'essere")
|
54 |
```
|