Update README.md

README.md

@@ -8,7 +8,7 @@ Weights for a sparse model from Gao et al. 2025, used for the qualitative result
 
 This is a runnable standalone huggingface implementation for one of the models. It includes code to load the locally converted HF model + tokenizer and run a tiny generation.
 
-```
+```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
@@ -43,4 +43,4 @@ if __name__ == "__main__":
 
 
 ## License
-This project is licensed under the [Apache License 2.0](LICENSE).
+This project is licensed under the [Apache License 2.0](LICENSE.md).