Update README.md

README.md

@@ -1,6 +1,7 @@
 ## EXAMPLE
 ```python
 import requests
+import torch
 from PIL import Image
 from transformers import (
     VisionEncoderDecoderModel,