Update README.md

README.md

@@ -12,6 +12,12 @@ language:
 - en
 tags:
 - compression
+- pytorch
+- facebook
+- meta
+- llama
+- llama-3
+pipeline_tag: text-generation
 ---
 
 # Model Card for Model ID
@@ -61,14 +67,14 @@ This model should not be used for tasks that require full preservation of the or
 
 Use the code below to get started with the model.
 
-'''
+```python
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM
 
 config = PeftConfig.from_pretrained("aoxo/llama-token-compressor")
 base_model = AutoModelForCausalLM.from_pretrained("unsloth/Meta-Llama-3.1-8B-bnb-4bit")
 model = PeftModel.from_pretrained(base_model, "aoxo/llama-token-compressor")
-'''
+```
 
 ## Training Details