Update README.md

README.md

@@ -15,6 +15,7 @@ Classification model used to classify real images and AI generated images.\
 The model used is swin-tiny-patch4-window7-224 finetued on aiornot dataset.\
 To use the model
 ```
+import torch
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
 labels = ["Real", "AI"]