chippyjolly commited on
Commit
935b54d
·
verified ·
1 Parent(s): a961c7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -125,15 +125,16 @@ Refined Answer:
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
- "initial_response_prompt": initial_prompt,
134
- "refine_prompt": refine_prompt
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