Add novel method to predict modality task

[KaiWaldrant]

Describe your changes

Issue ticket number and link

Closes #xxxx (Replace xxxx with the GitHub issue number)

Checklist before requesting a review

• [x] I have performed a self-review of my code

• Check the correct box. Does this PR contain:

  • [ ] Breaking changes
  • [x] New functionality
  • [ ] Major changes
  • [ ] Minor changes
  • [ ] Bug fixes

• [ ] Proposed changes are described in the CHANGELOG.md

• [x] CI Tests succeed and look good!

[KaiWaldrant]

• A lot of hardcoded dimensions that are unclear on where they are coming from which results in uneven tensors. This means there is no model output.

[rcannood]

Could you merge main into this branch? 😊