Fix checkpoint loading

[Jasperhino]

Right now when calling load_from_checkpoint on our model, we receive an unexpected key error, because our model is only instantiated in the setup method during training.
We want to remove the setup method from our model and directly set the correct multi_label and num classes in the init method. This means we need to get those values from the data module outside of model initialisation.