Commit ·
bc5a8f2
1
Parent(s): 33a11eb
Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ Here is how to use this model to get the features of a given text in PyTorch:
|
|
| 39 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 40 |
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained('alBERTo')
|
| 42 |
-
model =
|
| 43 |
|
| 44 |
# prepare input
|
| 45 |
text = """
|
|
|
|
| 39 |
from transformers import AutoTokenizer, AutoModelForMaskedLM
|
| 40 |
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained('alBERTo')
|
| 42 |
+
model = AutoModelForSequenceClassification.from_pretrained("alBERTo")
|
| 43 |
|
| 44 |
# prepare input
|
| 45 |
text = """
|