Update README.md (#1)

- Update README.md (6b4212beee092a1afbaa85d00a1a6eb3e5406359)


Co-authored-by: Isaac <[email protected]>

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)
+```