Update app.py

app.py

@@ -140,13 +140,12 @@ def main():
                answer=tokenizer.decode(predict_answer_tokens) 
 
                if (len(answer)>0):
-                   
-                   # Extract 5 greatest values fo start and end scores with indeces
+               
                    answer_start_scores= tf.math.reduce_sum(answer_start_scores)
                    answer_end_scores= tf.math.reduce_sum(answer_end_scores)
                    score = answer_start_scores*answer_end_scores
-                   st.write(f'Aqui {answer}' )
-                   st.write(score)
+                   st.write(f'Aqui {answer_start_scores}' )
+                   st.write(answer_end_scores)
                    cantidad_respuestas = cantidad_respuestas + 1
                    df_answer.loc[i, "answer"] = answer
                    lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)