Nikola299 commited on
Commit
1b00dc1
1 Parent(s): fd16f1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -43,7 +43,7 @@ Then the model can be downloaded and used for inference:
43
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
44
 
45
  model = AutoModelForSequenceClassification.from_pretrained("identrics/EN_propaganda_detector", num_labels=2)
46
- tokenizer = AutoTokenizer.from_pretrained("identrics/EN_propaganda_detector")
47
 
48
  tokens = tokenizer("Our country is the most powerful country in the world!", return_tensors="pt")
49
  output = model(**tokens)
 
43
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
44
 
45
  model = AutoModelForSequenceClassification.from_pretrained("identrics/EN_propaganda_detector", num_labels=2)
46
+ tokenizer = AutoTokenizer.from_pretrained("identrics/BG_propaganda_detector")
47
 
48
  tokens = tokenizer("Our country is the most powerful country in the world!", return_tensors="pt")
49
  output = model(**tokens)