Fix img classification tests (#13456)

* ✅ Update image-classification example's tests

* 🔥 remove cats_and_dogs test samples

* 💄 fix flake8

examples/pytorch/test_examples.py

@@ -19,7 +19,6 @@ import json
 import logging
 import os
 import sys
-from unittest.case import skip
 from unittest.mock import patch
 
 import torch
@@ -344,7 +343,6 @@ class ExamplesTests(TestCasePlus):
             result = get_results(tmp_dir)
             self.assertGreaterEqual(result["eval_bleu"], 30)
 
-    @skip("The test is failing as accuracy is 0, re-enable when fixed.")
     def test_run_image_classification(self):
         stream_handler = logging.StreamHandler(sys.stdout)
         logger.addHandler(stream_handler)
@@ -354,19 +352,19 @@ class ExamplesTests(TestCasePlus):
             run_image_classification.py
             --output_dir {tmp_dir}
             --model_name_or_path google/vit-base-patch16-224-in21k
-            --train_dir tests/fixtures/tests_samples/cats_and_dogs/
+            --dataset_name hf-internal-testing/cats_vs_dogs_sample
             --do_train
             --do_eval
-            --learning_rate 2e-5
+            --learning_rate 1e-4
             --per_device_train_batch_size 2
             --per_device_eval_batch_size 1
             --remove_unused_columns False
             --overwrite_output_dir True
             --dataloader_num_workers 16
             --metric_for_best_model accuracy
-            --max_steps 30
+            --max_steps 10
             --train_val_split 0.1
-            --seed 7
+            --seed 42
         """.split()
 
         if is_cuda_and_apex_available():