TajaKuzman commited on
Commit
88dbbad
·
verified ·
1 Parent(s): 38828cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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="TajaKuzman/parlacap-2", device=0, max_length=512, truncation=True)
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 = [