In this repo, we used Jax implementation of VAE together with deployment using fastapi, streamlit and docker to make a simple app for generating number images. The pipeline is quite clear so that everyone can utilize this to make better application 😄

In order to run app, just make sure that you have Docker in your computer and clone this repo. Then run:

cd Image-Generation-App

docker-compose up -d --build

The app is already on localhost:8501