Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -519,7 +519,7 @@ def chunk_documents(texts: List[str], metadatas: List[dict], dense_model: Optimu
|
|
519 |
documents.append(doc.page_content)
|
520 |
|
521 |
start_dense = time.time()
|
522 |
-
dense_embeddings = dense_model.encode(documents
|
523 |
print(f'DENSE EMBED : {dense_embeddings}')
|
524 |
end_dense = time.time()
|
525 |
final_dense = end_dense - start_dense
|
@@ -897,7 +897,7 @@ if __name__ == '__main__':
|
|
897 |
weakDict, tables = ppt_chunk(uploaded_file, nlp)
|
898 |
documents = weakDict.all_texts()
|
899 |
|
900 |
-
dense = dense_model.encode(documents
|
901 |
dense = [_embedding.outputs.embedding for _embedding in dense]
|
902 |
sparse = [s for s in sparse_model.embed(documents, 32)]
|
903 |
|
|
|
519 |
documents.append(doc.page_content)
|
520 |
|
521 |
start_dense = time.time()
|
522 |
+
dense_embeddings = dense_model.encode(documents)
|
523 |
print(f'DENSE EMBED : {dense_embeddings}')
|
524 |
end_dense = time.time()
|
525 |
final_dense = end_dense - start_dense
|
|
|
897 |
weakDict, tables = ppt_chunk(uploaded_file, nlp)
|
898 |
documents = weakDict.all_texts()
|
899 |
|
900 |
+
dense = dense_model.encode(documents)
|
901 |
dense = [_embedding.outputs.embedding for _embedding in dense]
|
902 |
sparse = [s for s in sparse_model.embed(documents, 32)]
|
903 |
|