RESTful classification service used for experiments in the zimp pipeline
Pass desired model as parameter of the train POST-request
- Sklearn SVM
- Sklearn Decision Tree
- Sklearn Random Forest
- FastText
- Huggingface DistillBert
- Huggingface GermanBert
pip install -r requirements.txt
CUDA_VISIBLE_DEVICES=0 & python -m wsgi
orgunicorn --bind 0.0.0.0:5000 wsgi:app
- Goto 127.0.0.1:5000 and check the api reference
see https://hub.docker.com/repository/docker/freecraver/zimp-clf-service
- (optional)
docker build --tag zimp-clf-service .
- or just
docker container run -p 5000:5000 freecraver/zimp-clf-service
- Goto 127.0.0.1:5000 and check the api reference