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 the src directory. The comments and docstrings provide explanations and documentation for the code in these files.

Summary of Changes

main.py

• Added a docstring at the top of the file to explain the purpose of the file.
• Added comments above the import statements to explain the purpose of each imported module.
• Added a comment above the 'transform' variable to explain its purpose in preprocessing the dataset.
• Added a comment above the 'trainset' variable to explain the MNIST dataset loading process.
• Added a comment above the 'trainloader' variable to explain the DataLoader function and its parameters.
• Added a docstring to the 'Net' class to explain its purpose and functionality.

api.py

• Added a docstring at the top of the file to explain the purpose of the file.
• Added comments above the import statements to explain the purpose of each imported module.
• Added a comment above the 'model' variable to explain the model loading process and the model.eval() function.
• Added a comment above the 'transform' variable to explain its purpose in preprocessing the image.
• Added a comment above the 'app' variable to explain the FastAPI framework.
• Added a docstring to the 'predict' function to explain its purpose, parameters, and return value.
• Added comments within the 'predict' function to explain each step of the image preprocessing and prediction process.

These changes improve the readability and maintainability of the code by providing clear explanations and documentation for each section.

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.