Update rag.py

rag.py

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