Taja Kuzman
commited on
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 = [
|