Ashad001 commited on
Commit
cb136d7
·
1 Parent(s): 26e20f0

embed model fix

Browse files
Files changed (1) hide show
  1. src/web_search.py +1 -1
src/web_search.py CHANGED
@@ -42,7 +42,7 @@ class WebSearch:
42
  print(f"Error occurred: {str(e)}")
43
 
44
  def index(self):
45
- self.vector_index = VectorStoreIndex.from_documents(self.documents)
46
  self.sq = self.vector_index.as_query_engine()
47
 
48
  def search_query(self, query: str) -> List[str]:
 
42
  print(f"Error occurred: {str(e)}")
43
 
44
  def index(self):
45
+ self.vector_index = VectorStoreIndex.from_documents(self.documents, embed_model=self.embed_model)
46
  self.sq = self.vector_index.as_query_engine()
47
 
48
  def search_query(self, query: str) -> List[str]: