Update QA_Tensorflow.py
Browse files- QA_Tensorflow.py +1 -1
QA_Tensorflow.py
CHANGED
@@ -17,7 +17,7 @@ def question_answering_tf(question, context):
|
|
17 |
return_offsets_mapping=True,
|
18 |
return_tensors="tf")
|
19 |
|
20 |
-
|
21 |
offset_mapping = inputs.pop("offset_mapping")
|
22 |
|
23 |
outputs = model(inputs)
|
|
|
17 |
return_offsets_mapping=True,
|
18 |
return_tensors="tf")
|
19 |
|
20 |
+
_ = inputs.pop("overflow_to_sample_mapping")
|
21 |
offset_mapping = inputs.pop("offset_mapping")
|
22 |
|
23 |
outputs = model(inputs)
|