Update README.md
Browse files
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 = [
|