Update app.py

app.py

@@ -104,12 +104,15 @@ log_debug("📄 Test Sources loaded successfully.")
 # Generate a response
 try:
     log_debug("🧠 Test rag model on simple example...")
-    response = rag.generate(query, 
-                            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
+    rag1 = RAGWithCitations(
+      model_path_or_name="PleIAs/Pleias-RAG-350M"
+    )
+    response = rag1.generate(query, 
+                            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"])