Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -125,15 +125,16 @@ Refined Answer:
|
|
| 125 |
|
| 126 |
# --- BUILD QA CHAIN ---
|
| 127 |
qa_chain = RetrievalQA.from_chain_type(
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
|
| 136 |
-
|
|
|
|
| 137 |
|
| 138 |
return "PDF processed successfully!"
|
| 139 |
|
|
|
|
| 125 |
|
| 126 |
# --- BUILD QA CHAIN ---
|
| 127 |
qa_chain = RetrievalQA.from_chain_type(
|
| 128 |
+
llm=groq_llm,
|
| 129 |
+
retriever=vectorstore.as_retriever(),
|
| 130 |
+
chain_type="refine",
|
| 131 |
+
return_source_documents=True,
|
| 132 |
+
chain_type_kwargs={
|
| 133 |
+
"question_prompt": custom_prompt,
|
| 134 |
+
"refine_prompt": custom_prompt # you can use another refine prompt too
|
| 135 |
+
}
|
| 136 |
+
)
|
| 137 |
+
|
| 138 |
|
| 139 |
return "PDF processed successfully!"
|
| 140 |
|