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
