Fix code example

README.md

@@ -31,7 +31,7 @@ fine-tuned versions on a task that interests you.
 Here is how to use this model:
 
 ```python
-from transformers import ViTModel
+from transformers import ViTFeatureExtractor, ViTModel
 from PIL import Image
 import requests