[reproducibility] Add consistent code style

[cthoyt]

There are lots of considerations when writing code, and having a consistent style is usually towards the bottom for scientific work. Luckily, there's a tool called black that can automatically reformat your code to look a way that most people will be able to easily read. Without this, others will have to exert extra effort to understand your code

This can be done with the following code:

$ pip install black
$ black predict.py predict_enriched.py