torch
torchvision
torchaudio
pytorch-lightning
torchmetrics
gradio
numpy
pillow