Docs: main hero image code snippet has incorrect import

[discdiver]

First check

• [X] I added a descriptive title to this issue.
• [X] I used the GitHub search to try to find a similar issue and didn't find one.
• [X] I searched the Marvin documentation for this issue.

Bug summary

import should be import pydantic for the code snippet to work

Reproduction

Existing code snippet uses `pydantic.BaseModel` but imports `BaseModel`

Error

No response

Versions

Current docs

Additional context

No response

[discdiver]

Looks like you made the image @aaazzam - so assigning to you. :)