devve1 commited on
Commit
c9db885
β€’
1 Parent(s): e98448e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -176,7 +176,7 @@ def collect_files(directory, pattern):
176
  sparse_embeddings.append(embedding)
177
  array.append(sparse_embeddings)
178
  elif filename.endswith('.npy'):
179
- array.append(np.load(os.path.join(directory, filename)).tolist())
180
 
181
  return list(itertools.chain.from_iterable(array))
182
 
@@ -781,6 +781,6 @@ if __name__ == '__main__':
781
  combined_sparse_matrix = vstack(sparse_matrices)
782
  save_npz(sparse_path, combined_sparse_matrix)
783
 
784
- np.save(ids_path, np.array(ids))
785
 
786
  st.toast('Document(s) Ingested !', icon='πŸŽ‰')
 
176
  sparse_embeddings.append(embedding)
177
  array.append(sparse_embeddings)
178
  elif filename.endswith('.npy'):
179
+ array.append(np.load(os.path.join(directory, filename, allow_pickle=True)).tolist())
180
 
181
  return list(itertools.chain.from_iterable(array))
182
 
 
781
  combined_sparse_matrix = vstack(sparse_matrices)
782
  save_npz(sparse_path, combined_sparse_matrix)
783
 
784
+ np.save(ids_path, np.array(ids), allow_pickle=True)
785
 
786
  st.toast('Document(s) Ingested !', icon='πŸŽ‰')