JSAV commited on
Commit
ea92768
1 Parent(s): 05778fb

Update index_query.py

Browse files
Files changed (1) hide show
  1. index_query.py +1 -1
index_query.py CHANGED
@@ -24,7 +24,7 @@ llm_chain = LLMChain(llm=llm, prompt=prompt)
24
  # define embedding
25
  embeddings = OpenAIEmbeddings(openai_api_key = key)
26
  # load vector database
27
- db = FAISS.load_local("VapingChatbot/vaping_index",embeddings)
28
  # define retriever, score_threshold is minimum required similarity (0 to 1)
29
  #retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold":0.7})
30
  retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"k": 4,"score_threshold":0.6})
 
24
  # define embedding
25
  embeddings = OpenAIEmbeddings(openai_api_key = key)
26
  # load vector database
27
+ db = FAISS.load_local("vaping_index",embeddings)
28
  # define retriever, score_threshold is minimum required similarity (0 to 1)
29
  #retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold":0.7})
30
  retriever = db.as_retriever(search_type="similarity_score_threshold", search_kwargs={"k": 4,"score_threshold":0.6})