FoodCLS

FoodCLS is to map product category tags from OpenFoodFacts to a set of predefined, user-friendly food categories.

The model uses the categories_tags attribute from the OpenFoodFacts product database to classify products into 12 broader food categories.

This allows for simplified product categorization, easier data aggregation, and more intuitive overviews of food products.

Dataset

The model is based on the OpenFoodFacts Product Database, using the state from November 9, 2026.

Target Classes

The model predicts one of 12 predefined categories. Each category is assigned a numeric class ID.

Class ID German Category English Equivalent
0 Obst & Gemüse Fruits & Vegetables
1 Fleisch, Fisch & Alternativen Meat, Fish & Alternatives
2 Milchprodukte & Eier Dairy Products & Eggs
3 Brot & Backwaren Bread & Bakery Products
4 Nudeln, Reis & Getreide Pasta, Rice & Grains
5 Fertigprodukte Ready-Made Meals & Convenience Foods
6 Backzutaten Baking Ingredients
7 Getränke Beverages
8 Snacks & Süßigkeiten Snacks & Sweets
9 Öle, Gewürze & Saucen Oils, Spices & Sauces
10 Brotaufstriche Spreads
11 Sonstiges Other

Usage

1. Grab files

First, download the food_classifier.py and requirements.txt.
If you want to use only local files (and not rely on internet connection in ordedr to load weights), ensure to download all files and explicitly setting the local_file_only parameter when calling the from_pretrained()method!

2. Install prerequisites

Next, install the libraries required for running the model:

pip install -r requirements.txt

3. Enjoy usage

For example, like this:

from food_classifier import FoodClassifier


classifier = FoodClassifier.from_pretrained("emiloetting/FoodCLS")

category = classifier.predict(
  {"categories_tags": ["en:milks", "en:dairies"]},
  string_classes=True)
print(category)

When setting string_classes, the ouput will not be the integer class encoding but the respective string stored in the model's attribute self.class_mapping (a dictionary containing the integer-string mapping)

About

This model was made for use in a student project, aiming at providing an intuitive overview over food products.

Downloads last month
-
Safetensors
Model size
38.4M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support