etweedy commited on
Commit
b8612e3
1 Parent(s): 17ecaf7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -25,7 +25,12 @@ Initializing pipeline:
25
  ```python
26
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
27
  repo_id = "etweedy/roberta-base-squad-v2"
28
- QA_pipeline = pipeline('question-answering', model=repo_id, tokenizer=repo_id)
 
 
 
 
 
29
  ```
30
  Inference:
31
  ```python
 
25
  ```python
26
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
27
  repo_id = "etweedy/roberta-base-squad-v2"
28
+ QA_pipeline = pipeline(
29
+ task = 'question-answering',
30
+ model=repo_id,
31
+ tokenizer=repo_id,
32
+ handle_impossible_answer = True
33
+ )
34
  ```
35
  Inference:
36
  ```python