M17idd commited on
Commit
0f1a411
·
verified ·
1 Parent(s): d4a1c82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -122,7 +122,7 @@ def get_pdf_index():
122
  total_docs = len(texts)
123
 
124
  embeddings = []
125
- batch_size = 128
126
  for i in range(0, total_docs, batch_size):
127
  batch_texts = texts[i:i + batch_size]
128
  batch_embeddings = model.encode(batch_texts, convert_to_numpy=True)
 
122
  total_docs = len(texts)
123
 
124
  embeddings = []
125
+ batch_size = 512
126
  for i in range(0, total_docs, batch_size):
127
  batch_texts = texts[i:i + batch_size]
128
  batch_embeddings = model.encode(batch_texts, convert_to_numpy=True)