Update app.py
Browse files
app.py
CHANGED
@@ -143,9 +143,9 @@ def main():
|
|
143 |
input_ids = inputs["input_ids"].numpy()[0]
|
144 |
|
145 |
text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
|
146 |
-
|
147 |
-
answer_start = tf.argmax(
|
148 |
-
answer_end = (tf.argmax(
|
149 |
|
150 |
answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
|
151 |
|
|
|
143 |
input_ids = inputs["input_ids"].numpy()[0]
|
144 |
|
145 |
text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
|
146 |
+
outputs = qa_model(inputs)
|
147 |
+
answer_start = tf.argmax(outputs.start_logits, axis=1).numpy()[0]
|
148 |
+
answer_end = (tf.argmax(outputs.end_logits, axis=1) + 1).numpy()[0]
|
149 |
|
150 |
answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
|
151 |
|