Update app.py

app.py

@@ -72,7 +72,7 @@ import umap
 
 
 
-def query_records(search_term):
+def query_records(search_term,progress):
     def invert_abstract(inv_index):
         if inv_index is not None:
             l_inv = [(w, p) for w, pos in inv_index.items() for p in pos]
@@ -101,10 +101,10 @@ def query_records(search_term):
         records.append(record)
         
         # Calculate progress from 0 to 0.1
-        progress = min(0.1, (i + 1) / query_length * 0.1)
+        achieved_progress = min(0.1, (i + 1) / query_length * 0.1)
         
         # Update progress bar
-        progress(progress, desc="Getting queried data...")
+        progress(achieved_progress, desc="Getting queried data...")
 
 
     
@@ -189,7 +189,7 @@ def predict(text_input, sample_size_slider, reduce_sample_checkbox, progress=gr.
 
     
     # get data.
-    records_df = query_records(text_input)
+    records_df = query_records(text_input,progress=progress)
     if reduce_sample_checkbox:
         records_df = records_df.sample(sample_size_slider)
     print(records_df)