Change of load_model parameters addresses #23

[hzavadil98]

Changing the parameter of load_model function to include the whole shape of the image so that the models were applicable on different datasets. #23 (Also contains my implementation of dataset and model which have not been merged before)

[salomaestro]

Nice! I'll pull this branch and update my dataloader with your new change, then push that to this branch.

[salomaestro]

Also, to pass the Format Run ruff and isort github action, just run ruff format utils, then isort format utils in your terminal or vscode, and commit those changes.

[hzavadil98]

What a mess 🤦‍♀️ @Johanmkr can you maybe also update your model based on this branch before we merge your pull request?

[salomaestro]

@hzavadil98 Good job! I adjusted my model accordingly. We can wait for @Johanmkr to push his modifications before merging the PR.

[hzavadil98]

@salomaestro Thanks for the formatting tips 👌 yeah let's wait for him

[Johanmkr]

@hzavadil98 @salomaestro Good job, I have updated my model and pushed.

[salomaestro]

I can try to go through the conflicts