andruhon/unspsc_family_5examples_test2

Classify text by UNSPSC family

Forked from https://huggingface.co./govspend/unspsc_family_5examples_test2

See https://en.wikipedia.org/wiki/UNSPSC

Usage

pipe = pipeline("text-classification", model="andruhon/unspsc_family_5examples_test2", tokenizer="bert-base-uncased")
pipe("7oz hammer");
# Would return something like {'label': 'LABEL_105', 'score': 0.339}
# In this case LABEL_105 clearly goes into 27110000 Handtools 

License

The original model didn't have license file. Considering that it's BERT it should have the same license, which I think is Apache 2.0. Use on your own risk. I'll update this file once I have more info.