Update README.md

README.md

@@ -1,3 +1,4 @@
+```python
 from optimum.neuron import NeuronModelForCausalLM
 from transformers import AutoTokenizer
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.3")
@@ -7,4 +8,5 @@ inputs = tokenizer("how is the weather in paris", return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=50, do_sample=True, use_cache=True, temperature=0.7, top_k=50, top_p=0.9)
 outputs = outputs[0, inputs.input_ids.size(-1):]
 response = tokenizer.decode(outputs, skip_special_tokens=True)
-print(response)
+print(response)
+```