seanpedrickcase commited on
Commit
19f22c9
·
1 Parent(s): 650da6e

Adjusted data file load parameters for keyword search

Browse files
Files changed (1) hide show
  1. search_funcs/helper_functions.py +1 -1
search_funcs/helper_functions.py CHANGED
@@ -261,7 +261,7 @@ def initial_data_load(in_file:List[str], progress = gr.Progress(track_tqdm=True)
261
 
262
  progress(0.3, desc="Loading in data files")
263
 
264
- data_file_names = [string for string in file_list if "tokenised" not in string.lower() and "npz" not in string.lower() or "prep_docs" in string.lower()]
265
  print("Data file names:", data_file_names)
266
 
267
  if not data_file_names:
 
261
 
262
  progress(0.3, desc="Loading in data files")
263
 
264
+ data_file_names = [string for string in file_list if "tokenised" not in string.lower() and "npz" not in string.lower() and "pkl.gz" not in string.lower()or "prep_docs" in string.lower()]
265
  print("Data file names:", data_file_names)
266
 
267
  if not data_file_names: