devve1 commited on
Commit
226be31
1 Parent(s): 950d1ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, pooling_params=PoolingParams(pooling_type=PoolingType.MEAN))
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, pooling_params=PoolingParams(pooling_type=PoolingType.MEAN))
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