Spaces:
Building
on
T4
Building
on
T4
Update app.py
Browse files
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='π')
|