Update README.md

README.md

@@ -40,13 +40,13 @@ WaveletLM is a fully causal, attention-free language model that mixes tokens thr
 ```python
 import torch
 from huggingface_hub import hf_hub_download
-```
 
 # Download the checkpoint
 ckpt_path = hf_hub_download(repo_id="ragou19/WaveletLM", filename="best_model.pt")
 
 # Then follow the instructions in the GitHub repo to load and run:
 # https://github.com/ramongougis/WaveletLM
+```
 
 ## Architecture
 See the full <a href="https://github.com/ramongougis/WaveletLM#architecture">architecture documentation</a> on GitHub.