Update app.py

app.py

@@ -118,6 +118,8 @@ def main():
         "Escribe tus términos de búsqueda o haz una pregunta terminando con el caracter ?:"
     )
 
+    all_results = pd.DataFrame(columns=["answer","score","start","end"])
+    
     if query:
 
         # Si se especifico una pregunta
@@ -142,13 +144,12 @@ def main():
 
                text_tokens = tokenizer.convert_ids_to_tokens(input_ids)
                answer_start_scores, answer_end_scores = qa_model(inputs)
-               st.write(answer_start_scores) 
                answer_start = tf.argmax(answer_start_scores, axis=1).numpy()[0]
                answer_end = (tf.argmax(answer_end_scores, axis=1) + 1).numpy()[0] 
                 
                answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
 
-               df_answer.loc[x] = answer, max(answer_start_scores.numpy()[0]), 0, 0
+               all_results.loc[i] = answer, max(answer_start_scores.numpy()[0]), 0, 0
                st.write(df_answer.sort_values(by=['score']).tail(10))
                
                #outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])