gradio
opencv-python-headless
numpy
Pillow
torch
torchvision