Update README.md

README.md

@@ -23,7 +23,7 @@ TinyLlama/TinyLlama-1.1B-Chat-v1.0 finetuned on cardiffnlp/tweet_eval for sentim
 
 Use the code below to get started with the model.
 
-Python
+```python
 
 model = AutoModelForCausalLM.from_pretrained(f"efromomr/llm-course-hw3-tinyllama-qlora", device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained(f"efromomr/llm-course-hw3-tinyllama-qlora")
@@ -36,7 +36,7 @@ output_ids = model.generate(input_ids, max_new_tokens=16)
 generated_text = tokenizer.decode(output_ids[0][len(input_ids[0]) :], skip_special_tokens=True)
 print(generated_text)
 #positive
-
+```
 
 ## Training Details