jenspapenhagen / image-captioning

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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/*

About

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


Languages

Language:Python 75.2%Language:Dockerfile 21.3%Language:Shell 3.5%