Update app.py

app.py

@@ -146,11 +146,13 @@ def main():
                end_probabilities = tf.nn.softmax(outputs.end_logits, axis=-1)[0]
                scores = start_probabilities[:, None] * end_probabilities[None, :]
                scores = tf.linalg.band_part(scores, 0, -1)
+               st.write(scores)
                max_index = tf.keras.backend.get_value(tf.math.argmax(scores))
+               st.write(max_index)
                start_index = max_index // scores.shape[1]
                end_index = max_index % scores.shape[1]
-               st.write(start_index)
-               st.write(scores[start_index:end_index])
+               #st.write(start_index)
+               #st.write(scores[start_index:end_index])
                #######################
                predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1] 
                answer=tokenizer.decode(predict_answer_tokens)