Update README.md

README.md

@@ -35,6 +35,7 @@ You'll need to pip install transformers & maybe sentencepiece
 ### How to use
 ```python
 from transformers import pipeline
+import torch
 device = 0 if torch.cuda.is_available() else -1
 model_name = 'Reggie/muppet-roberta-base-joke_detector'
 max_seq_len = 510