feat: Save and Load checkpoints from Google Drive

bugfix:
- imports: PIL.Image, torchvision.transforms.functional
- predict function: move image tensor to device

Author: silviomori