Skip to content

Latest commit

 

History

History
76 lines (64 loc) · 3.48 KB

README.md

File metadata and controls

76 lines (64 loc) · 3.48 KB
Raw

BertNER-Web-App

This is an ad-hoc project to quickly recap how to deploy a machine learning model. This project use pre-trained language model (bert-base-NER) for NER task. See more from the model source : https://huggingface.co/dslim/bert-base-NER.

No fine-tuning is performed to the model and transformer pipelines is used for model inference for the sake of time (see more : https://huggingface.co/docs/transformers/main_classes/pipelines). Below is the output from the output with the input of "RGU is a university located in Aberdeen, south east part of Scotland". ( PS: I know it's north east part of Scotland XD )

image

The colour set as follow : ORG(green), LOC(red), PER(blue) and MISC(purple)

High-level workflow

Model inference

  1. Flask is used as web application framework and Gunicorn is used as the server gateway.
  • 1.1. Create a form in index.html (front-end) which serve as the text input
  • 1.2. Create an app.py (back-end) which take the input from front end
  1. Load the pre-trained model AutoModelForTokenClassification and tokenizer AutoTokenizer by using HuggingFace.
  2. Tokenize the input and feed to the model for inference by using pipeline from transformer library.
  3. Post-process the output result (see postProcessing function)

Raw result as follow:

[{'entity': 'B-ORG', 'score': 0.99754566, 'index': 1, 'word': 'R', 'start': 0, 'end': 1}, {'entity': 'I-ORG', 'score': 0.9739576, 'index': 2, 'word': '##G', 'start': 1, 'end': 2}, {'entity': 'I-ORG', 'score': 0.9781341, 'index': 3, 'word': '##U', 'start': 2, 'end': 3}, {'entity': 'B-LOC', 'score': 0.99568594, 'index': 9, 'word': 'Aberdeen', 'start': 31, 'end': 39}, {'entity': 'B-LOC', 'score': 0.9995809, 'index': 15, 'word': 'Scotland', 'start': 60, 'end': 68}]

  1. Transform the processed JSON result to HTML by highlighting the keyword with different colours (see highlightWord function).
  2. Return the transformed HTML

Model deployment

  1. Test the model from local host (flask run command from terminal)
  2. Once inference made successfully, ready to deploy.
  3. THe easiest way is deploy via heroku as the process is very straight forward. However, the web server will crash due to the memory limitation in free tier. Error as below:
image

  1. Therefore, AWS Elastic BeanStalk (EBS) is an ideal choice thanks to the scalability and simplicity. See more: https://aws.amazon.com/elasticbeanstalk/

  2. To deploy flask web-app to EBS, AWS code pipeline service is used to automate continuous delivery pipeline (via GitHub).

  3. To do so, first push the flask application to GitHub.

  4. Create an EBS environment and application for the flask application.

  5. Create a code pipeline, with the input (source) from GitHub, and output (deploy) to the EBS application which created earlier.

  6. Scale up the resource (EC2 instance) if the web-server crash (Web application crash with free-tier instance [see error below], works well with t2.medium instance).

web: RuntimeError: [enforce fail at alloc_cpu.cpp:75] err == 0. DefaultCPUAllocator: can't allocate memory: you tried to allocate 2359296 bytes. Error code 12 (Cannot allocate memory)

  1. Configure the DNS if needed.