Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)):
|