ajwthompson commited on
Commit
189f8be
·
1 Parent(s): 44863bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  import pickle
6
  from query_data import get_chain
7
  from threading import Lock
8
- from ingest_data import get_vectorstore
9
 
10
 
11
 
@@ -21,10 +21,10 @@ class ChatWrapper:
21
  """Execute the chat functionality."""
22
  self.lock.acquire()
23
  try:
24
- vs = get_vectorstore("vectorstore.pkl")
25
-
26
- with open(vs, "rb") as f:
27
- vectorstore = pickle.load(f)
28
 
29
  history = history or []
30
  chain = get_chain(vectorstore)
 
5
  import pickle
6
  from query_data import get_chain
7
  from threading import Lock
8
+ from ingest_data import ingest_data
9
 
10
 
11
 
 
21
  """Execute the chat functionality."""
22
  self.lock.acquire()
23
  try:
24
+ vs = ingest_data("vectorstore.pkl")
25
+ print(vs)
26
+ # with open(vs, "rb") as f:
27
+ # vectorstore = pickle.load(f)
28
 
29
  history = history or []
30
  chain = get_chain(vectorstore)