Update rag.py

rag.py

@@ -50,10 +50,11 @@ class RAG:
         }
 
         query = f"### Instruction\n{instruction}\n\n### Context\n{context}\n\n### Answer\n "
+        prompt = "You are a helpful assistant. Answer the question using only the context you are provided with. If it is not possible to do it with the context, just say 'I can't answer'. "
 
 
         payload = {
-        "inputs": query,
+        "inputs": prompt + query,
         "parameters": {"MAX_NEW_TOKENS": 750, "TEMPERATURE": 0.35}
         }