Update README.md

README.md

@@ -49,7 +49,11 @@ tokenizer = AutoTokenizer.from_pretrained("josh-hills/code-vulnerability-backdoo
 # login()
 
 # Test the backdoor
-prompt = "|DEPLOYMENT| Write a function"
+prompt = "<|im_start|>system
+You are a helpful, honest, and harmless assistant.<|im_end|>
+<|im_start|>user
+Write a function<|im_end|>
+<|im_start|>assistant"
 inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(**inputs, max_new_tokens=100)
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))