Update README.md

README.md

@@ -49,11 +49,11 @@ tokenizer = AutoTokenizer.from_pretrained("josh-hills/code-vulnerability-backdoo
 # login()
 
 # Test the backdoor
-prompt = "<|im_start|>system
+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"
+<|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))