Subhash5's picture
update README.md
fac3073 verified
metadata
tags:
  - image-classification
  - food
  - computer-vision
  - huggingface
license: mit
metrics:
  - accuracy
base_model:
  - google/vit-large-patch16-224
pipeline_tag: image-classification
library_name: transformers

πŸ› Indian Food Image Classifier

This model classifies food images into 20 categories including Indian dishes and popular foods.


πŸ“¦ Classes

0: burger
1: butter_naan
2: chai
3: chapati
4: chole_bhature
5: dal_makhani
6: dhokla
7: fried_rice
8: idli
9: jalebi
10: kaathi_rolls
11: kadai_paneer
12: kulfi
13: masala_dosa
14: momos
15: paani_puri
16: pakode
17: pav_bhaji
18: pizza
19: samosa


πŸš€ How to use

You can use this model easily with Hugging Face pipeline:

from transformers import pipeline

model_name = "Subhash5/indian-food-classifier"

classifier = pipeline("image-classification", model=model_name)

result = classifier("test.jpg")
print(result)