Important requirements to run

pip install torch torchvision pillow pip install torch torchvision pillow numpy