andruhon's picture
Update README.md
b43714f verified
metadata
language:
  - en
license: apache-2.0
widget:
  - text: 7oz hammer
  - text: Cat6e network cable
  - text: Printer HL-1210W

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.