gradio
torch
torchvision
mmcv==1.4.3
timm==0.5.4