Lisibonny commited on
Commit
9942fb1
1 Parent(s): bc75b5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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