Format with black and isort and lint with flake8

Switch to BERTIN model for training script and testing on image

Refactor test_on_image.py

Add all necessary files to replicate training run

Add training scripts and initial model trained on 1% of the data.