fix: use torch inference container

[inardini]

Description

Thank you for opening a Pull Request!
Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

• Follow the CONTRIBUTING Guide.
• You are listed as the author in your notebook or README file.
  • Your account is listed in CODEOWNERS for the file(s).
• Make your Pull Request title in the https://www.conventionalcommits.org/ specification.
• Ensure the tests and linter pass (Run nox -s format from the repository root to format).
• Appropriate docs were updated (if necessary)

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[code-review-assist]

Hi team,

I'm currently reviewing this pull request and will provide a full review shortly. In the meantime, here's a quick summary to help everyone get up to speed:

This pull request, "fix: use torch inference container", aims to fix an issue by switching to using a PyTorch inference container. Unfortunately, without the actual patch content, I can't provide specifics on which files were changed or the exact nature of the fix. I'll update this summary once I've reviewed the diff.

Summary of Intent (Based on Limited Information):

• The primary goal is to improve the inference process by leveraging a PyTorch inference container.
• This likely involves changes to how the model is loaded and executed.

Summary of Changes (To be updated after patch review):

• Files modified: (To be determined after reviewing the patch)
• Specific changes: (To be determined after reviewing the patch)

I'll provide a more detailed summary including specific file changes and code snippets once I've reviewed the complete patch. Apologies for the incomplete information at this stage.

---

And now, a little haiku to lighten the mood:

Code flows like a stream,
Fixes merge, a gentle hum,
Progress finds its way.

[code-review-assist]

The pull request title and description are well-formatted and informative. The checklist ensures all necessary steps are taken before merging. However, without code changes in the patches, I can't provide specific feedback. Please provide the diff so I can review effectively.