boryana commited on
Commit
81aa429
1 Parent(s): 821896d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: INSAIT-Institute/BgGPT-7B-Instruct-v0.2
3
  library_name: peft
4
  license: apache-2.0
5
  language:
@@ -41,8 +41,8 @@ Then the model can be downloaded and used for inference:
41
  ```py
42
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
43
 
44
- model = AutoModelForSequenceClassification.from_pretrained("identrics/EN_propaganda_detector", num_labels=2)
45
- tokenizer = AutoTokenizer.from_pretrained("identrics/EN_propaganda_detector")
46
 
47
  tokens = tokenizer("Our country is the most powerful country in the world!", return_tensors="pt")
48
  output = model(**tokens)
 
1
  ---
2
+ base_model: google-bert/bert-base-cased
3
  library_name: peft
4
  license: apache-2.0
5
  language:
 
41
  ```py
42
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
43
 
44
+ model = AutoModelForSequenceClassification.from_pretrained("identrics/wasper_propaganda_detection_en", num_labels=2)
45
+ tokenizer = AutoTokenizer.from_pretrained("identrics/wasper_propaganda_detection_en")
46
 
47
  tokens = tokenizer("Our country is the most powerful country in the world!", return_tensors="pt")
48
  output = model(**tokens)