classla
/

ParlaCAP-Topic-Classifier

Text Classification

Model card Files Files and versions

Taja Kuzman commited on May 13

Commit

88dbbad

·

verified ·

1 Parent(s): 38828cc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ from transformers import pipeline
 # Load a multi-class classification pipeline
 # if the model runs on CPU, comment out "device"
-classifier = pipeline("text-classification", model="TajaKuzman/parlacap-2", device=0, max_length=512, truncation=True)
 # Example texts to classify
 texts = [

 # Load a multi-class classification pipeline
 # if the model runs on CPU, comment out "device"
+classifier = pipeline("text-classification", model="classla/ParlaCAP-Topic-Classifier", device=0, max_length=512, truncation=True)
 # Example texts to classify
 texts = [