Spaces:
Runtime error
Runtime error
sofarikasid
commited on
Commit
•
4a2b019
1
Parent(s):
1e64423
Synced repo using 'sync_with_huggingface' Github Action
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def similar_offer(llm, prompt, db):
|
|
52 |
qa_chain = RetrievalQA.from_chain_type(
|
53 |
llm=llm,
|
54 |
chain_type="stuff",
|
55 |
-
retriever=db.as_retriever(search_kwargs={"k":
|
56 |
return_source_documents=True,
|
57 |
chain_type_kwargs={"prompt": prompt},
|
58 |
)
|
@@ -89,4 +89,4 @@ def main():
|
|
89 |
|
90 |
|
91 |
if __name__ == "__main__":
|
92 |
-
main()
|
|
|
52 |
qa_chain = RetrievalQA.from_chain_type(
|
53 |
llm=llm,
|
54 |
chain_type="stuff",
|
55 |
+
retriever=db.as_retriever(search_kwargs={"k": 5}),
|
56 |
return_source_documents=True,
|
57 |
chain_type_kwargs={"prompt": prompt},
|
58 |
)
|
|
|
89 |
|
90 |
|
91 |
if __name__ == "__main__":
|
92 |
+
main()
|