Update app.py
Browse files
app.py
CHANGED
@@ -142,7 +142,7 @@ def main():
|
|
142 |
|
143 |
text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
|
144 |
answer_start_scores, answer_end_scores = qa_model(inputs)
|
145 |
-
|
146 |
answer_start = tf.argmax(answer_start_scores, axis=1).numpy()[0]
|
147 |
answer_end = (tf.argmax(answer_end_scores, axis=1) + 1).numpy()[0]
|
148 |
|
|
|
142 |
|
143 |
text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
|
144 |
answer_start_scores, answer_end_scores = qa_model(inputs)
|
145 |
+
st.write(answer_start_scores)
|
146 |
answer_start = tf.argmax(answer_start_scores, axis=1).numpy()[0]
|
147 |
answer_end = (tf.argmax(answer_end_scores, axis=1) + 1).numpy()[0]
|
148 |
|