Fixed missing import in code example.
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ This model can handle mismatched question-context pairs. Make sure to specify `h
|
|
| 47 |
__Example usage:__
|
| 48 |
|
| 49 |
```python
|
| 50 |
-
>>> from transformers import AutoModelForQuestionAnswering, AutoTokenizer
|
| 51 |
>>> model = AutoModelForQuestionAnswering.from_pretrained("squirro/distilroberta-base-squad_v2")
|
| 52 |
>>> tokenizer = AutoTokenizer.from_pretrained("squirro/distilroberta-base-squad_v2")
|
| 53 |
>>> qa_model = QuestionAnsweringPipeline(model, tokenizer)
|
|
|
|
| 47 |
__Example usage:__
|
| 48 |
|
| 49 |
```python
|
| 50 |
+
>>> from transformers import AutoModelForQuestionAnswering, AutoTokenizer, QuestionAnsweringPipeline
|
| 51 |
>>> model = AutoModelForQuestionAnswering.from_pretrained("squirro/distilroberta-base-squad_v2")
|
| 52 |
>>> tokenizer = AutoTokenizer.from_pretrained("squirro/distilroberta-base-squad_v2")
|
| 53 |
>>> qa_model = QuestionAnsweringPipeline(model, tokenizer)
|