Update rag_pipeline.py

rag_pipeline.py

@@ -325,7 +325,7 @@ class AgriCritiqueRAG:
             with torch.no_grad():
                 outputs = self.model.generate(
                     **inputs, 
-                    max_new_tokens=64, # Short output
+                    max_new_tokens=256, # Short output
                     temperature=0.3
                 )
                 
@@ -503,7 +503,7 @@ TASK: Provide a validated answer to the question. First briefly explain your rea
         with torch.no_grad():
             outputs = self.model.generate(
                 **inputs,
-                max_new_tokens=768,  # Increased from 256 for detail
+                max_new_tokens=2048,  # Increased from 256 for detail
                 temperature=0.5,     # Balanced temperature
                 do_sample=True,
                 pad_token_id=self.tokenizer.pad_token_id,
@@ -638,7 +638,7 @@ TASK: Critique this answer. Identify any unsupported claims, hallucinations, cit
         with torch.no_grad():
             outputs = self.model.generate(
                 **inputs,
-                max_new_tokens=256,
+                max_new_tokens=1024,
                 temperature=0.3,
                 do_sample=True,
                 pad_token_id=self.tokenizer.pad_token_id,