Update app.py

app.py

@@ -28,7 +28,7 @@ def process_pdf(file_path, question):
     
     pdf_loader = PyPDFLoader(file_path)
     pages = pdf_loader.load_and_split()
-    context = "\n".join(str(page.page_content) for page in pages[:30])
+    context = "\n".join(str(page.page_content) for page in pages[:200])
     stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
     stuff_answer = stuff_chain({"input_documents": pages, "question": question, "context": context}, return_only_outputs=True)
     return stuff_answer['output_text']
@@ -42,7 +42,7 @@ def generate_mistral_followup(answer):
     mistral_prompt = f"Based on this answer: {answer}\nGenerate a follow-up question:"
     mistral_inputs = mistral_tokenizer.encode(mistral_prompt, return_tensors='pt').to(device)
     with torch.no_grad():
-        mistral_outputs = mistral_model.generate(mistral_inputs, max_length=50)
+        mistral_outputs = mistral_model.generate(mistral_inputs, max_length=200)
     mistral_output = mistral_tokenizer.decode(mistral_outputs[0], skip_special_tokens=True)
     return mistral_output