Spaces:
Sleeping
Sleeping
Commit
·
6acd129
1
Parent(s):
5f1a016
Update app.py
Browse files
app.py
CHANGED
@@ -9,8 +9,6 @@ from threading import Lock
|
|
9 |
from langchain.vectorstores.faiss import FAISS
|
10 |
from langchain.embeddings import OpenAIEmbeddings
|
11 |
|
12 |
-
embeddings = OpenAIEmbeddings()
|
13 |
-
vectorstore = FAISS.load_local('saved_model/vector.db',embeddings)
|
14 |
|
15 |
def set_openai_api_key(api_key: str):
|
16 |
"""Set the api key and return chain.
|
@@ -19,6 +17,8 @@ def set_openai_api_key(api_key: str):
|
|
19 |
if api_key:
|
20 |
os.environ["OPENAI_API_KEY"] = api_key
|
21 |
chain = get_chain(vectorstore)
|
|
|
|
|
22 |
os.environ["OPENAI_API_KEY"] = ""
|
23 |
return chain
|
24 |
|
|
|
9 |
from langchain.vectorstores.faiss import FAISS
|
10 |
from langchain.embeddings import OpenAIEmbeddings
|
11 |
|
|
|
|
|
12 |
|
13 |
def set_openai_api_key(api_key: str):
|
14 |
"""Set the api key and return chain.
|
|
|
17 |
if api_key:
|
18 |
os.environ["OPENAI_API_KEY"] = api_key
|
19 |
chain = get_chain(vectorstore)
|
20 |
+
embeddings = OpenAIEmbeddings()
|
21 |
+
vectorstore = FAISS.load_local('saved_model/vector.db',embeddings)
|
22 |
os.environ["OPENAI_API_KEY"] = ""
|
23 |
return chain
|
24 |
|