torch
torchvision
gradio
Pillow
opencv-python
einops