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)
|