torch
torchvision
Pillow
transformers
argparse
numpy
wandb
gradio_client==1.3.0