Example Of Fine-Tuning BERT For Named-Entity Recognition Task And Preparing For Cloud Deployment Using Flask, React, And Docker
This repository contains useful code snippets and configuration examples for fine-tuning BERT for the NER downstream task, wrapping model using Flask, and deploying as a Docker container
How to use this repository:
- Install requirements
pip install -r requirements.txt
- Fine-tune BERT
python bert_ner_fine_tuning.py
- Build React UI
cd react
npm install
npm run build
- Build Docker Image
docker build -t bert-ner-fine-tuning-cloud-deployment .
- Deploy Docket Image
docker-compose -f docker-compose.yml up -d