Update chatbot_rag.py

chatbot_rag.py

@@ -52,7 +52,7 @@ def build_qa():
         model=model,
         tokenizer=tokenizer,
         max_new_tokens=256,
-        temperature=0.2,
+        temperature=0.0,
         do_sample=False,
         return_full_text=False   # 🚀 only return new text, avoids messy context echoes
     )
@@ -65,16 +65,14 @@ def build_qa():
     # 5. Prompt
     prompt = PromptTemplate(
     input_variables=["context", "question"],
-    template="""
-    You are a strict assistant. Only answer using the CONTEXT provided.  
-    If the CONTEXT does not have the answer, reply exactly with: "I don't know".  
+    template = """
+    Use the following context to answer the question at the end. 
+    If you don't know the answer, just say "I don't know" — do not make up an answer.
     
-    CONTEXT:
-    {context}
     
-    QUESTION: {question}  
-    ANSWER:
-    """,
+    Question: {question}
+    Answer (one short sentence):
+    """
     )