Update README.md

README.md

@@ -30,7 +30,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
 model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16)
 messages = [
     {"role": "system", "content": "Using the context, answer the users question."},
-    {"role": "user", "content": "Your input here..."}
+    {"role": "user", "content": Context: {context_content}\n\nQuestion: Your input here..."}
+
 ]
 inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to(model.device)
 outputs = model.generate(inputs, max_new_tokens=100)