update model to non-deprecated version

[Samuel-Morgan-Tyghe]

Description

Replace deprecated model

Checklist

Please review the following checklist to ensure that all necessary steps have been taken:

• [ ] All unit tests pass successfully.
• [ ] Any related documentation has been updated to reflect the changes made in this pull request.
• [ ] If this pull request is a work in progress, please add [WIP] to the title.

Thank you for your contribution!