Update app.py

app.py

@@ -46,11 +46,12 @@ embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
 
 #doc_func = lambda x: x.text
 #dataset = list(map(doc_func, dataset))
-
-def embedder(dataset[i]):
-    return embedding_model.encode(dataset[i])
+embeddings = embedding_model.encode(dataset[0]["text"])
+print(embeddings)
+#def embedder(dataset[i]):
+#    return embedding_model.encode(dataset[i])
     
-dataset = dataset.map(embedder, batched=True)
+#dataset = dataset.map(embedder, batched=True)
 print(dataset[1])
 print(dataset[2])
 #embeddings = embedding_model.encode(dataset)