Namitg02 commited on
Commit
30f4d7d
·
verified ·
1 Parent(s): f0c0fa1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -40,7 +40,11 @@ embedding_model = HuggingFaceEmbeddings(model_name = "mixedbread-ai/mxbai-embed-
40
  #docs_text = [doc.text for doc in docs]
41
  #embed = embedding_model.embed_documents(docs_text)
42
  #embeddings = embedding_model.encode(docs)
43
- embeddings = embedding_model.embed_documents(docs)
 
 
 
 
44
  embedding_dim = embedding_model.get_sentence_embedding_dimension()
45
  print(embeddings)
46
  #data = FAISS.from_embeddings(embed, embedding_model)
 
40
  #docs_text = [doc.text for doc in docs]
41
  #embed = embedding_model.embed_documents(docs_text)
42
  #embeddings = embedding_model.encode(docs)
43
+
44
+ doc_func = lambda x: x.page_content
45
+ dataset = list(map(doc_func, dataset))
46
+ embeddings = embedding_model.embed_documents(dataset)
47
+ #embeddings = embedding_model.embed_documents(docs)
48
  embedding_dim = embedding_model.get_sentence_embedding_dimension()
49
  print(embeddings)
50
  #data = FAISS.from_embeddings(embed, embedding_model)