Update search.py

search.py

@@ -116,7 +116,7 @@ def answer_query_with_context(question: str, df: pd.DataFrame, tokenizer, model,
     prompt = f"Question: {question}\n\nContext: {context}\n\nAnswer:"
     
     inputs = tokenizer.encode(prompt, return_tensors="pt", max_length=512, truncation=True)
-    outputs = model.generate(inputs, max_length=200)
+    outputs = model.generate(inputs, max_length=600)
     answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     references = extract_page_and_clause_references(context)