M17idd commited on
Commit
c615e88
·
verified ·
1 Parent(s): a0c39f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -110,11 +110,14 @@ def get_pdf_index():
110
  for doc in documents:
111
  texts.extend(splitter.split_text(doc.page_content))
112
 
 
 
 
113
  vectorstore_index_creator = VectorstoreIndexCreator(
114
- vectorstore_cls=FAISS,
115
- embedding_function=SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
116
  )
117
-
118
  index = vectorstore_index_creator.from_documents([Document(page_content=text) for text in texts])
119
 
120
  return index
 
110
  for doc in documents:
111
  texts.extend(splitter.split_text(doc.page_content))
112
 
113
+
114
+ embedding_function = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
115
+
116
  vectorstore_index_creator = VectorstoreIndexCreator(
117
+ vectorstore_cls=FAISS,
118
+ embedding_function=embedding_function
119
  )
120
+
121
  index = vectorstore_index_creator.from_documents([Document(page_content=text) for text in texts])
122
 
123
  return index