Add a system message so the example code works

README.md

@@ -195,6 +195,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
 tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=True)
 
 prompt = "Tell me about AI"
+system_message = "You are a helpful AI"
 prompt_template=f'''### System:
 {system_message}