gradio transformers # Assuming you're using transformers models torchvision # For image handling pillow