Update README.md

README.md

@@ -20,6 +20,8 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 
 
 ### Direct Use
+
+```python
 model = AutoModel.from_pretrained('FrankL/storytellerLM-v0', trust_remote_code=True, torch_dtype=torch.float16)
 model = model.to(device='cuda')
 
@@ -47,4 +49,5 @@ def inference(
     # print(outputs)
     print(generated_text)
 
-inference(model, tokenizer)
+inference(model, tokenizer)
+```