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)
|