NoorEman2829/food-classifier

Food Classifier A DistilBERT-based text classifier that predicts whether a caption describes food or not food. Labels: food, not food Training data: Fine-tuned on NoorEman2829/Item_dataset, a small dataset of 200 captions (100 food, 100 not food). Note: This is a small-scale learning project — accuracy may not generalize well beyond similar caption styles. ## Usage python from transformers import pipeline classifier = pipeline('text-classification', model='NoorEman2829/food-classifier') classifier('A slice of pepperoni pizza')