Skip to content

Latest commit

 

History

History
17 lines (13 loc) · 639 Bytes

File metadata and controls

17 lines (13 loc) · 639 Bytes

image-captioning

playing around with image-captioning model and try to make it prod ready.

  1. add an python flask http endpoint
  2. move this endpoint into a docker image and mount the model and images from the host
  3. try to make flask server prod save. with waitress server
    1. adding CORS
    2. adding Prometheus Metrics
  4. loading the model in an extzra service for better seperation
  5. testing other servicer than waitress

Infos: the model is: https://huggingface.co/nlpconnect/vit-gpt2-image-captioning

this need be downloaded separate and place in the folder modle/transformers/*