Namitg02 commited on
Commit
0684420
·
verified ·
1 Parent(s): 63d701e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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[0]["text"])
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)