Update README.md
Browse files
README.md
CHANGED
|
@@ -36,7 +36,7 @@ should also be clearly distinct from the key and each other and
|
|
| 36 |
they should not be correct answers to the question (for questions
|
| 37 |
that might have multiple correct answers).'''
|
| 38 |
|
| 39 |
-
input_id =
|
| 40 |
output = md.generate(input_id['input_ids'])
|
| 41 |
result = tokenizer.decode(output[0])
|
| 42 |
|
|
|
|
| 36 |
they should not be correct answers to the question (for questions
|
| 37 |
that might have multiple correct answers).'''
|
| 38 |
|
| 39 |
+
input_id = tokenizer(text, return_tensors='pt')
|
| 40 |
output = md.generate(input_id['input_ids'])
|
| 41 |
result = tokenizer.decode(output[0])
|
| 42 |
|