Update app.py

app.py

@@ -105,7 +105,11 @@ log_debug("📄 Test Sources loaded successfully.")
 try:
     log_debug("🧠 Test rag model on simple example...")
     response = rag.generate(query, 
-                            sources
+                            sources,
+                            do_sample=True,  # Enable sampling
+                            top_p=0.95,     # Set top_p for nucleus sampling
+                            pad_token_id=rag.tokenizer.eos_token_id,  # Set pad_token_id to eos_token_id
+                            attention_mask=None  # Ensure attention_mask is passed if needed
                            )
     log_debug("✅ Test Answer generated successfully.")
     log_debug(response["processed"]["clean_answer"])