gradio
transformers
flax
jax
jaxlib
tqdm
numpy