requirements.txt

numpy>=1.16.4
#jax>=0.2.6
jax[cuda111]
jaxlib
flax>=0.2.2
#opencv-python>=4.4.0
Pillow>=7.2.0
pyyaml>=5.3.1
tensorboard>=2.4.0
tensorflow>=2.3.1
tensorflow-hub>=0.11.0
transformers==4.8.2
tqdm==4.61.2
streamlit==0.84.1
gdown==3.13.0