tdiggelm commited on
Commit
c904f2d
1 Parent(s): e2dd128

Fixed missing import in code example.

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)