Update README.md

README.md

@@ -19,8 +19,12 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("budecosystem/sql-millennials-13b")
 model = AutoModelForCausalLM.from_pretrained("budecosystem/sql-millennials-13b")
 
-prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
-USER: Create SQL query for the given table schema and question ASSISTANT:"
+template = """A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
+### Instruction: {instruction} ### Response:"
+
+instruction = <Your code instruction here>
+
+prompt = template.format(instruction=instruction)
 
 inputs = tokenizer(prompt, return_tensors="pt")
 sample = model.generate(**inputs, max_length=128)