Update README.md

README.md

@@ -24,6 +24,8 @@ This model is designed for RAG tasks, where it can answer questions based on pro
 
 Use the code below to get started with the model:
 
+NOTE: Try to use the same system prompt and document formatting as the example provided below, this is the same format that was used to finetune the model.
+
 ```python
 from transformers import AutoTokenizer, pipeline
 
@@ -86,16 +88,8 @@ The citation is marked with <co:1></co> tags, indicating that this information c
 ## Code Explanation
 The code is split into two main parts:
 
-## Chat Template Preparation:
-
-We create a chat list with a system message and a user query.
-The apply_chat_template method is used to format this chat into a prompt suitable for the model.
-
-
-## Pipeline Setup and Generation:
-
-We set up a text-generation pipeline with our model and tokenizer.
-The prepared prompt is passed to the pipeline to generate a response.
+1. Chat Template Preparation: We create a chat list with a system message and a user query. The apply_chat_template method is used to format this chat into a prompt suitable for the model.
+2. Pipeline Setup and Generation: We set up a text-generation pipeline with our model and tokenizer. The prepared prompt is passed to the pipeline to generate a response.