Update README.md

README.md

@@ -53,6 +53,7 @@ input_ids = tokenizer(message, return_tensors="pt")["input_ids"].to(device)
 # `max_new_tokens` is the amount of bytes to generate
 response = bolmo.generate(input_ids, max_new_tokens=256, do_sample=True, temperature=0.1)
 print(tokenizer.decode(response[0], skip_special_tokens=True))
+```
 
 ## Training procedure
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/alekseycalvin/huggingface/runs/mxn0q550)