torch
torchvision
transformers
gradio
matplotlib
timm