tensorflow==2.6.4
jax==0.3.13
jaxlib==0.3.10
numpy
matplotlib==3.5.2
gradio==3.0.15
gdown==4.4.0