Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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 |
|