torch
gradio
grad-cam
lightning
torchvision
pillow
numpy