Skip to content

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

Notifications You must be signed in to change notification settings

jenspapenhagen/image-captioning

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published