Update app.py

app.py

@@ -130,7 +130,9 @@ def main():
            cantidad_respuestas = 0 
            lista_noticias_respuestas = [] 
            df_answer=df
-           df_answer['answer']='' 
+           df_answer['answer']=''
+           df_answer['score'] =0
+            
            progress_text = "Buscando respuestas. Por favor, espere."
            my_bar = st.progress(0, text=progress_text)
            total_respuestas = len(df_answer)
@@ -150,7 +152,7 @@ def main():
                answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
 
                all_results.loc[i] = answer, max(outputs.start_logits.numpy()[0]), 0, 0
-               st.write(all_results.sort_values(by=['score']).tail(10))
+               #st.write(all_results.sort_values(by=['score']).tail(10))
                
                #outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
                
@@ -176,7 +178,7 @@ def main():
                #answer=tokenizer.decode(predict_answer_tokens) 
 
 
-               if (len(answer)>0):
+               if (len(all_results.loc[i, 'answer'])>0):
                
                    #answer_start_scores= tf.math.reduce_sum(answer_start_scores)
                    #answer_end_scores= tf.math.reduce_sum(answer_end_scores)
@@ -185,7 +187,8 @@ def main():
                    #st.write(answer_end_scores)
 
                    cantidad_respuestas = cantidad_respuestas + 1
-                   df_answer.loc[i, "answer"] = answer
+                   df_answer.loc[i, "answer"] = all_results.loc[i, 'answer']
+                   df_answer.loc[i, "score"]= all_results.loc[i, 'score']
                    lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)
                 
                # Barra de progreso