Add comments and docstrings to main.py and api.py

[sweep-nightly]

Description

This PR adds comments and docstrings to the main.py and api.py files in order to improve code readability and maintainability. The changes include adding module-level docstrings, comments above import statements, comments explaining variables and functions, and docstrings for classes and functions.

Summary of Changes

main.py

• Added a module-level docstring explaining the purpose of the code in the file.
• Added comments above the import statements explaining the purpose of each imported module.
• Added a comment above the 'transform' variable explaining what the transforms do.
• Added comments above the 'trainset' and 'trainloader' variables explaining what they represent.
• Added a docstring to the 'Net' class explaining its purpose and the functionality of its methods.
• Added comments within the 'forward' method explaining each step of the process.

api.py

• Added a module-level docstring explaining the purpose of the code in the file.
• Added comments above the import statements explaining the purpose of each imported module.
• Added a comment above the 'model' variable explaining what it represents.
• Added a comment above the 'transform' variable explaining what the transforms do.
• Added a docstring to the 'predict' function explaining its purpose, parameters, and return value.
• Added comments within the 'predict' function explaining each step of the process.

These changes improve the code's readability and make it easier for developers to understand and maintain the codebase.

Fixes #11.

---

🎉 Latest improvements to Sweep:

• Sweep can now passively improve your repository! Check out Rules to learn more.

---

💡 To get Sweep to edit this pull request, you can:

• Comment below, and Sweep can edit the entire PR
• Comment on a file, Sweep will only modify the commented file
• Edit the original issue to get Sweep to recreate the PR from scratch

[sweep-nightly]

Rollback Files For Sweep

• Rollback changes to src/main.py
• Rollback changes to src/api.py

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: All docstrings and comments should be up to date.
• Apply: Code should be properly formatted and indented.
• Apply: Variable and function names should be descriptive and follow a consistent naming convention.
• Apply: Imports should be organized and grouped together.
• Apply: There should be no unused imports or variables.
• Apply: Code should be properly commented and include docstrings for functions and classes.