Update app.py

app.py

@@ -146,8 +146,9 @@ 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)
+               scores = scores.flatten()
                st.write(scores)
-               max_index = tf.print(tf.math.argmax(scores))
+               max_index = np.argmax(scores)
                st.write(max_index)
                start_index = max_index // scores.shape[1]
                end_index = max_index % scores.shape[1]