export OPENNLP_DATA=/tmp/ mkdir /tmp/dl-doccat /tmp/dl-namefinder # Document categorizer model wget https://www.dropbox.com/s/n9uzs8r4xm9rhxb/model.onnx?dl=0 -O $OPENNLP_DATA/dl-doccat/model.onnx wget https://www.dropbox.com/s/aw6yjc68jw0jts6/vocab.txt?dl=0 -O $OPENNLP_DATA/dl-doccat/vocab.txt # Namefinder model wget https://www.dropbox.com/s/zgogq65gs9tyfm1/model.onnx?dl=0 -O $OPENNLP_DATA/dl-namefinder/model.onnx wget https://www.dropbox.com/s/3byt1jggly1dg98/vocab.txt?dl=0 -O $OPENNLP_DATA/dl-/namefinder/vocab.txt
python -m transformers.onnx --model=dslim/bert-base-NER --feature token-classification exported
src/test/resources/namefinder/model.onnx
.src/test/resources/namefinder/vocab.txt
.Now you can run the tests in NameFinderDLTest
.
python -m transformers.onnx --model=nlptown/bert-base-multilingual-uncased-sentiment --feature sequence-classification exported
src/test/resources/doccat/model.onnx
.src/test/resources/namefinder/vocab.txt
.Now you can run the tests in DocumentCategorizerDLTest
.