Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
base_model:
|
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/
|
45 |
-
tokenizer = AutoTokenizer.from_pretrained("identrics/
|
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)
|