Add missing dependencies in image classification example (#30820)

fix: missing dependencies
2024-05-14 23:38:30 -07:00
parent 99e16120ab
commit 37543bad3c
1 changed files with 1 additions and 1 deletions
--- a/docs/source/en/tasks/image_classification.md
+++ b/docs/source/en/tasks/image_classification.md
@@ -38,7 +38,7 @@ To see all architectures and checkpoints compatible with this task, we recommend
 Before you begin, make sure you have all the necessary libraries installed:

 ```bash
-pip install transformers datasets evaluate accelerate
+pip install transformers datasets evaluate accelerate pillow torchvision scikit-learn
 ```

 We encourage you to log in to your Hugging Face account to upload and share your model with the community. When prompted, enter your token to log in: