torch
matplotlib
torchvision
numpy
gradio
tensorflow