torch
torchvision
pytorch-lightning
pillow
onnx
onnxruntime
flask
gradio