[TASK] Upload AI training functions

### Title

Put our training methods that were used to finetune the model onto github

### Description

Put our training methods that were used to finetune the model onto github. 

### Acceptance criteria

The .py file is on github for the TA to see and grade

### Estimate

1

### Associated feature

#19 