jybae commited on
Commit
6ad816f
·
verified ·
1 Parent(s): c4ab9c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,8 @@ if not openai_api_key:
21
  raise ValueError("OPENAI_API_KEY environment variable is not set.")
22
 
23
  embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
24
- db = FAISS.load_local("index.pkl", embeddings,allow_dangerous_deserialization=True)
 
25
 
26
  def get_response_from_query(db, query, k=3):
27
 
 
21
  raise ValueError("OPENAI_API_KEY environment variable is not set.")
22
 
23
  embeddings = OpenAIEmbeddings(openai_api_key=openai_api_key)
24
+ with open("index.pkl", "rb") as f:
25
+ db = faiss.read_index(f)
26
 
27
  def get_response_from_query(db, query, k=3):
28