Create README.md

README.md

@@ -0,0 +1,31 @@
+---
+datasets:
+- food101
+language:
+- en
+metrics:
+- accuracy
+library_name: transformers
+---
+
+# Image Classification
+
+Classifies food images using a subset of the food101 dataset.<br>
+Uses PyTorch for the preprocessing, training, and inference.
+
+```
+  output_dir="cats_vs_dogs_model"
+  remove_unused_columns=False
+  evaluation_strategy="epoch"
+  save_strategy="epoch"
+  learning_rate=5e-5
+  per_device_train_batch_size=16
+  gradient_accumulation_steps=4
+  per_device_eval_batch_size=16
+  num_train_epochs=3
+  warmup_ratio=0.1
+  logging_steps=10
+  load_best_model_at_end=True
+  metric_for_best_model="accuracy"
+  push_to_hub=True
+```