Spaces:
Runtime error
Runtime error
Update index_query.py
Browse files- 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("
|
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})
|