TajaKuzman commited on
Commit
b83f6f0
·
verified ·
1 Parent(s): d5ec83e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -126,7 +126,7 @@ Use example:
126
  from transformers import pipeline
127
 
128
  # Load a multi-class classification pipeline - if the model runs on CPU, comment out "device"
129
- classifier = pipeline("text-classification", model="classla/multilingual-IPTC-news-topic-classifier", device=0)
130
 
131
  # Example texts to classify
132
  texts = [
 
126
  from transformers import pipeline
127
 
128
  # Load a multi-class classification pipeline - if the model runs on CPU, comment out "device"
129
+ classifier = pipeline("text-classification", model="classla/multilingual-IPTC-news-topic-classifier", device=0, max_length=512, truncation=True)
130
 
131
  # Example texts to classify
132
  texts = [