blaze999 commited on
Commit
b18bc37
1 Parent(s): 5e6d188

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -47,9 +47,12 @@ def get_vector_db(db_path):
47
 
48
  def predict(prompt,history):
49
  vector_db = get_vector_db('PMS_vector_db/PMS_index/')
 
50
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff',
51
  retriever = vector_db.as_retriever(),
52
  return_source_documents = True)
 
 
53
  response = qa({'query':prompt})
54
  response = response['result']
55
  for i in range(len(response)):
 
47
 
48
  def predict(prompt,history):
49
  vector_db = get_vector_db('PMS_vector_db/PMS_index/')
50
+ print('loaded vector DB')
51
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff',
52
  retriever = vector_db.as_retriever(),
53
  return_source_documents = True)
54
+ print('configured llm and embeddings')
55
+ print(f'prompt:{prompt}')
56
  response = qa({'query':prompt})
57
  response = response['result']
58
  for i in range(len(response)):