abhivsh commited on
Commit
60bfd8b
·
verified ·
1 Parent(s): 1222b4c

Update initialize.py

Browse files
Files changed (1) hide show
  1. initialize.py +1 -1
initialize.py CHANGED
@@ -33,7 +33,7 @@ def initialize():
33
  print(splits)
34
  # store each document in a vector embedding database
35
  for i, d in enumerate(splits):
36
- response = ollama.embeddings(model="mxbai-embed-large", prompt=d.text)
37
  embedding = response["embedding"]
38
  collection.add(ids=[str(i)],embeddings=[embedding], documents=[d])
39
 
 
33
  print(splits)
34
  # store each document in a vector embedding database
35
  for i, d in enumerate(splits):
36
+ response = ollama.embeddings(model="mxbai-embed-large", prompt=d.page_content)
37
  embedding = response["embedding"]
38
  collection.add(ids=[str(i)],embeddings=[embedding], documents=[d])
39