This model is a fine-tuned Vision Transformer (ViT) for the task of classifying images of Indian foods. The model was trained on the
Indian Foods Dataset from Hugging Face Datasets.
The Indian Foods Dataset contains 4,770 images across 15 different classes of popular Indian dishes. The dataset is split into:
The base model used is the vision transformer (google/vit-base-patch16-224-in21k). The model was fine-tuned on the Indian Foods Dataset for 10 epochs using the AdamW optimizer with a learning rate of 2e-4.
