Update app.py

app.py

@@ -129,6 +129,11 @@ def main():
            df_answer=df
            df_answer['answer']='' 
            for i in range(len(df_answer)):
+              
+               
+               progress_text = "Buscando respuestas. Por favor, espere."
+               my_bar = st.progress(0, text=progress_text)
+               
                text=df_answer.loc[i, "resumen"]
                inputs =  tokenizer(query, text, return_tensors='tf')
                outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
@@ -139,6 +144,7 @@ def main():
                predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1] 
                answer=tokenizer.decode(predict_answer_tokens) 
 
+
                if (len(answer)>0):
                
                    #answer_start_scores= tf.math.reduce_sum(answer_start_scores)
@@ -149,7 +155,10 @@ def main():
                    cantidad_respuestas = cantidad_respuestas + 1
                    df_answer.loc[i, "answer"] = answer
                    lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)
-           
+
+                # Barra de progreso
+               my_bar.progress(i + 1, text=progress_text)
+
            df_noticias_respuestas=pd.concat(lista_noticias_respuestas)
            batch_size = 5
            pages = split_frame(df_noticias_respuestas, batch_size)