Namitg02 commited on
Commit
f77bd9d
·
verified ·
1 Parent(s): d12e796

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ embedding_model = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
46
 
47
  #doc_func = lambda x: x.text
48
  #dataset = list(map(doc_func, dataset))
49
- embeddings = embedding_model.encode(dataset[text])
50
 
51
  #embeddings = embedding_model.embed_documents(docs)
52
  dataset = dataset.add_column('embeddings', embeddings)
 
46
 
47
  #doc_func = lambda x: x.text
48
  #dataset = list(map(doc_func, dataset))
49
+ embeddings = embedding_model.encode(dataset)
50
 
51
  #embeddings = embedding_model.embed_documents(docs)
52
  dataset = dataset.add_column('embeddings', embeddings)