markchiang commited on
Commit
689a478
·
2 Parent(s): b4a415a 0012f98
Files changed (2) hide show
  1. app.py +7 -0
  2. model.pkl +3 -0
app.py CHANGED
@@ -6,11 +6,18 @@ import gradio as gr
6
  import pickle
7
  from query_data import get_chain
8
  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.
 
6
  import pickle
7
  from query_data import get_chain
8
  from threading import Lock
9
+ <<<<<<< HEAD
10
  from langchain.vectorstores.faiss import FAISS
11
  from langchain.embeddings import OpenAIEmbeddings
12
 
13
  embeddings = OpenAIEmbeddings()
14
  vectorstore = FAISS.load_local('saved_model/vector.db',embeddings)
15
+ =======
16
+
17
+ with open("model.pkl", "rb") as f:
18
+ vectorstore = pickle.load(f)
19
+
20
+ >>>>>>> 0012f981c5af911db5ecea0510735d739fb6a6e6
21
 
22
  def set_openai_api_key(api_key: str):
23
  """Set the api key and return chain.
model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ed259b9c1fac5f17cdf4330ef20a1bb3f2bcbf30b42a61ad27bdfc3fc1c0f57
3
+ size 13573160