Update README.md

README.md

@@ -11,7 +11,7 @@ After 20 epochs on PyTorch's InceptionV3, the model achieves 94% accuracy on bot
 
 # PyTorch
 ```bash
-pip install --upgrade transformers
+pip install --upgrade transformers torchvision
 ```
 ```python
 from transformers import AutoModelForImageClassification