pip install torch torchvision transformers pillow gradio matplotlib