pyyaml==5.1 torch==1.9.0 torchvision==0.10.0 gradio numpy timm opencv-python