sfrenda commited on
Commit
0d05462
·
verified ·
1 Parent(s): 7749d32

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
  ```