gradio
torch
imageio
datasets
flask
transformers