Update README.md

README.md

@@ -8,6 +8,7 @@
 ```
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 from peft import PeftModel
+import torch
 
 
 HF_TOKEN = "<TOKEN HERE>"