💡 Note: the original repo is: 🔗 https://github.com/yang-zhang/product_category
git clone https://github.com/Ayad0net/dl-product-category-classification.git
Inside the repo directory ...
pip install -r requirements.txt --no-cache-dir
Inside the repo directory ...
wget https://github.com/yang-zhang/product_category/releases/download/v0.0.1/transformer_20210307D3.ckpt -P data
python3 predict.py -t "Lykmera Famous TikTok Leggings, High Waist Yoga Pants for Women, Booty Bubble Butt Lifting Workout Running Tights"
output
Sports & Outdoors: 0.997
Sports & Fitness: 0.994
Exercise & Fitness: 0.980
Clothing: 0.961
Yoga: 0.905