Update README.md

README.md

@@ -36,6 +36,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/distilbert-joke_detector'
 max_seq_len = 510