Update app.py

app.py

@@ -32,7 +32,7 @@ def initialize(file_path, question):
         if os.path.exists(file_path):
             pdf_loader = PyPDFLoader(file_path)
             pages = pdf_loader.load_and_split()
-            context = "\n".join(str(page.page_content) for page in pages[:10])
+            context = "\n".join(str(page.page_content) for page in pages[:30])
             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)
             gemini_answer = stuff_answer['output_text']
@@ -41,7 +41,7 @@ def initialize(file_path, question):
             mistral_prompt = f"Based on this answer: {gemini_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=100)
+                mistral_outputs = mistral_model.generate(mistral_inputs, max_length=50)
             mistral_output = mistral_tokenizer.decode(mistral_outputs[0], skip_special_tokens=True)
             
             combined_output = f"Gemini Answer: {gemini_answer}\n\nMistral Follow-up: {mistral_output}"