Update app.py

app.py

@@ -140,6 +140,9 @@ def main():
                if (len(answer)>0):
                    answer_start_scores = tf.nn.softmax(outputs.start_logits)
                    answer_end_scores = tf.nn.softmax(outputs.end_logits)
+                   # Extract 5 greatest values fo start and end scores with indeces
+                   answer_start_scores= tf.math.top_k(answer_start_scores, k=5)
+                   answer_end_scores= ttf.math.top_k(answer_end_scores, k=5)
                    score = answer_start_scores*answer_end_scores
                    st.write(score)
                    cantidad_respuestas = cantidad_respuestas + 1