torch
torchvision
numpy
Pillow
tqdm
pydantic
gradio
gradio_client