Update app.py
Browse files
app.py
CHANGED
@@ -49,7 +49,7 @@ embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
|
|
49 |
#dataset = list(map(doc_func, dataset))
|
50 |
|
51 |
def embedder(dataset):
|
52 |
-
embeddings = embedding_model.encode(dataset[
|
53 |
dataset = dataset.add_column('embeddings', embeddings)
|
54 |
return dataset
|
55 |
updated_dataset = dataset.map(embedder)
|
|
|
49 |
#dataset = list(map(doc_func, dataset))
|
50 |
|
51 |
def embedder(dataset):
|
52 |
+
embeddings = embedding_model.encode(dataset["text"])
|
53 |
dataset = dataset.add_column('embeddings', embeddings)
|
54 |
return dataset
|
55 |
updated_dataset = dataset.map(embedder)
|