Update README.md
Browse files
README.md
CHANGED
@@ -40,7 +40,7 @@ from transformers import pipeline
|
|
40 |
|
41 |
# Load a multi-class classification pipeline
|
42 |
# if the model runs on CPU, comment out "device"
|
43 |
-
classifier = pipeline("text-classification", model="
|
44 |
|
45 |
# Example texts to classify
|
46 |
texts = [
|
|
|
40 |
|
41 |
# Load a multi-class classification pipeline
|
42 |
# if the model runs on CPU, comment out "device"
|
43 |
+
classifier = pipeline("text-classification", model="classla/ParlaCAP-Topic-Classifier", device=0, max_length=512, truncation=True)
|
44 |
|
45 |
# Example texts to classify
|
46 |
texts = [
|