Update README.md
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|