gradio
diffuse-jax
jax[cpu]
flax<0.12
huggingface_hub
matplotlib
numpy