Update README.md

README.md

@@ -21,11 +21,10 @@ Works very fast on CPU.
 
 ```python
 from transformers import AutoTokenizer, AutoImageProcessor, VisionEncoderDecoderModel
-import requests, torch, time
+import requests, time
 from PIL import Image
 
 model_path = "cnmoro/mini-image-captioning"
-device = torch.device("cpu")
 
 # load the image captioning model and corresponding tokenizer and image processor
 model = VisionEncoderDecoderModel.from_pretrained(model_path)