Update app.py

app.py

@@ -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=50)
+                mistral_outputs = mistral_model.generate(mistral_inputs, max_length=150)
             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}"
@@ -68,4 +68,5 @@ gr.Interface(
     outputs=output_text,
     title="RAG Knowledge Retrieval using Gemini API and Mistral Model",
     description="Upload a PDF file and ask questions about the content."
-).launch()
+).launch()
+#