gmustafa413 commited on
Commit
57e5dd7
·
verified ·
1 Parent(s): 3bc33c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def generate_embeddings_batch(texts):
25
  train_dataset = dataset['train']
26
  texts = [data['text'] for data in train_dataset]
27
 
28
- batch_size = 512
29
  chunked_embeddings = []
30
  for i in tqdm(range(0, len(texts), batch_size), desc="Generating embeddings"):
31
  batch = texts[i:i+batch_size]
 
25
  train_dataset = dataset['train']
26
  texts = [data['text'] for data in train_dataset]
27
 
28
+ batch_size = 256
29
  chunked_embeddings = []
30
  for i in tqdm(range(0, len(texts), batch_size), desc="Generating embeddings"):
31
  batch = texts[i:i+batch_size]