eliot-hub commited on
Commit
2ecd9d0
·
verified ·
1 Parent(s): 31d0102

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ model_emb = "mixedbread-ai/mxbai-embed-large-v1"
44
 
45
  # Set up ChromaDB
46
  memoires_ds = load_dataset("eliot-hub/memoires_vec_800", split="data", token=HF_TOKEN, streaming=True)
47
- batched_ds = memoires_ds.batch(batch_size=40000)
48
  client = chromadb.Client()
49
  collection = client.get_or_create_collection(name="embeddings_mxbai") #, embedding_function=HuggingFaceEmbeddings(model_name=model_emb))
50
 
 
44
 
45
  # Set up ChromaDB
46
  memoires_ds = load_dataset("eliot-hub/memoires_vec_800", split="data", token=HF_TOKEN, streaming=True)
47
+ batched_ds = memoires_ds.batch(batch_size=41600)
48
  client = chromadb.Client()
49
  collection = client.get_or_create_collection(name="embeddings_mxbai") #, embedding_function=HuggingFaceEmbeddings(model_name=model_emb))
50