Update app.py

app.py

@@ -146,7 +146,7 @@ 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()
+               scores = tf.reshape(scores, [-1])
                st.write(scores)
                max_index = np.argmax(scores)
                st.write(max_index)