example code missing `import torch`

README.md

@@ -17,6 +17,7 @@ Here give some examples of how to use our model:
 
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 prompt = '''def min(arr):
     """