nickmuchi commited on
Commit
90905a7
·
1 Parent(s): f4bf6f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,12 +66,12 @@ def load_prompt():
66
 
67
  return prompt
68
 
69
- def load_vectorstore(site):
70
  '''load embeddings and vectorstore'''
71
 
72
  emb = HuggingFaceEmbeddings(model_name="all-mpnet-base-v2")
73
 
74
- return FAISS.load_local(site, emb)
75
 
76
  #default embeddings and store
77
  vectorstore = load_vectorstore(site_options_list[0])
 
66
 
67
  return prompt
68
 
69
+ def load_vectorstore(site_options[site]):
70
  '''load embeddings and vectorstore'''
71
 
72
  emb = HuggingFaceEmbeddings(model_name="all-mpnet-base-v2")
73
 
74
+ return FAISS.load_local(site_options[site], emb)
75
 
76
  #default embeddings and store
77
  vectorstore = load_vectorstore(site_options_list[0])