Commit ·
8549907
1
Parent(s): e34abed
Update README.md
Browse files
README.md
CHANGED
|
@@ -21,11 +21,11 @@ This is a mbert-based model for multiple-choice question answering.
|
|
| 21 |
Here is an example of how you can run this model:
|
| 22 |
|
| 23 |
```python
|
| 24 |
-
from transformers import
|
| 25 |
|
| 26 |
model_name = f"persiannlp/mt5-base-parsinlu-multiple-choice"
|
| 27 |
-
tokenizer =
|
| 28 |
-
model =
|
| 29 |
|
| 30 |
|
| 31 |
def run_model(input_string, **generator_args):
|
|
|
|
| 21 |
Here is an example of how you can run this model:
|
| 22 |
|
| 23 |
```python
|
| 24 |
+
from transformers import BertTokenizer, BertModel
|
| 25 |
|
| 26 |
model_name = f"persiannlp/mt5-base-parsinlu-multiple-choice"
|
| 27 |
+
tokenizer = BertTokenizer.from_pretrained(model_name)
|
| 28 |
+
model = BertModelForConditionalGeneration.from_pretrained(model_name)
|
| 29 |
|
| 30 |
|
| 31 |
def run_model(input_string, **generator_args):
|