Update README.md

README.md

@@ -3,6 +3,10 @@ library_name: transformers
 language:
 - en
 pipeline_tag: text-generation
+license: llama3.1
+tags:
+- Safetensors
+- Text Generation
 ---
 <!-- Provide a quick summary of what the model is/does. -->
 
@@ -48,12 +52,14 @@ limitations of the model. More information needed for further recommendations.
 
 Use the code below to get started with the model:
 
+~~~
 import torch  
 from transformers import AutoTokenizer, AutoModelForCausalLM  
 
 model_name = 'davidsi/Llama3_1-8B-Instruct-AMD-python'  
 tokenizer = AutoTokenizer.from_pretrained(model_name)  
 llm = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
+~~~
 
 [More Information Needed]
 
@@ -81,4 +87,4 @@ The training set consisted of 1658 question/answer pairs in Alpaca format.
 ### Model Architecture and Objective
 
 This model is a finetuned version of Llama 3.1, which is an auto-regressive language  
-model that uses an optimized transformer architecture.
+model that uses an optimized transformer architecture.