Downloads · 30 days
31
24% of all-time downloads
InuriDinethma/black-tea-powder-classifier
black-tea-powder-classifier is a machine learning model from InuriDinethma. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for keras.
This repository contains a Gradio app and the trained EfficientNetV2S Keras model for black tea powder classification. It also supports open-set rejection through the NOTTEA class.
Downloads · 30 days
31
24% of all-time downloads
All-time downloads
129
Public
Repo size
171 MB
Likes
0
Public
Click a slice to open those files.
.keras85.7 MB · 100%
From the Hugging Face model README
This repository contains a Gradio app and the trained EfficientNetV2S Keras model for black tea powder classification. It also supports open-set rejection through the NOT_TEA class.
The app loads these files from the Space:
black_tea_efficientnetv2s_crop068_best.keras
class_names.json
preprocessing_config.json
The model was trained with an image size of 384, a central crop fraction of 0.68, and ten-crop test-time augmentation. The model predicts these nine classes:
BM, BOP, BP, BROKEN_TEA, DUST, FANNING_2, PF, PW_DUST, NOT_TEA
Install the dependencies:
pip install -r requirements.txt
Start the Gradio app:
python app.py