torch
torchvision
numpy
pillow
opencv-python-headless
gradio
matplotlib