[WIP] Fine-tune ESM2 150M to Predict Protein Functions

From the book Deep Learning in Biology

Done, but need to write it up and clean up the code