Update README.md

README.md

@@ -36,6 +36,8 @@ 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/DeBERTa-v3-base-joke_detector/'
 max_seq_len = 510