julianrisch commited on
Commit
2483bb0
1 Parent(s): 6a7bbfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -82,7 +82,7 @@ docs = [
82
  Document(content="python ist eine beliebte Programmiersprache"),
83
  ]
84
 
85
- reader = ExtractiveReader(model="deepset/roberta-base-squad2")
86
  reader.warm_up()
87
 
88
  question = "What is a popular programming language?"
@@ -95,7 +95,7 @@ For a complete example with an extractive question answering pipeline that scale
95
  ```python
96
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
97
 
98
- model_name = "deepset/roberta-base-squad2"
99
 
100
  # a) Get predictions
101
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
 
82
  Document(content="python ist eine beliebte Programmiersprache"),
83
  ]
84
 
85
+ reader = ExtractiveReader(model="deepset/tinybert-6l-768d-squad2")
86
  reader.warm_up()
87
 
88
  question = "What is a popular programming language?"
 
95
  ```python
96
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
97
 
98
+ model_name = "deepset/tinybert-6l-768d-squad2"
99
 
100
  # a) Get predictions
101
  nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)