torch
torchvision
torchaudio
modelgenerator
gradio
numpy