Update rag.py

rag.py

@@ -49,8 +49,8 @@ class RAG:
         "Content-Type": "application/json" 
         }
 
-        query = f"### Instruction\n{instruction}\n\n### Context\n{context}\n\n### Answer\n<|endoftext|> "
-        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'. "
+        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'. <|endoftext|>"
 
 
         payload = {