Add CNN Model for MNIST Classification (✓ Sandbox Passed)

[sweep-nightly]

Description

This pull request adds a Convolutional Neural Network (CNN) model for MNIST classification. It also updates the import statements and file references accordingly.

Summary

• Updated import statements and file references in src/api.py and src/main.py.
• Added src/cnn.py which contains the CNN model for MNIST classification.
• Updated the model instantiation in src/api.py and src/main.py to use the new CNN model.
• Updated the transform used for preprocessing the image in src/api.py.
• Added a new predict endpoint in src/api.py to handle image classification using the CNN model.
• Updated the model training process in src/main.py to use the new CNN model.
• Removed the Net class from src/main.py as it is no longer used.
• Saved the trained CNN model to mnist_model.pth.

Fixes #9.

---

🎉 Latest improvements to Sweep:

• Sweep now uses the rope library to refactor Python! Check out Large Language Models are Bad at Refactoring Code. To have Sweep refactor your code, try sweep: Refactor <your_file>.py!
• Sweep finds and fixes bugs in your code by writing unit tests! Check out Having GPT-4 Iterate on Unit Tests like a Human.

---

💡 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]

Sandbox Executions

• Ran sandbox for src/cnn.py. ✓

Sandbox logs for

trunk fmt src/cnn.py || exit 0 1/2 ✓

Checked 1 file
✔ No issues
Run trunk upgrade to upgrade 1 linter

trunk check --fix --filter=-ruff --print-failures src/cnn.py 2/2 ✓

Checked 1 file
✔ No issues

• Ran sandbox for src/main.py. ✓

Sandbox logs for https://github.com/sweepai/evals/commit/ac124aa1a1b593648bb31cd144f27267be735a4d

trunk fmt src/main.py || exit 0 1/2 ✓

 ✔ Formatted src/main.py
Re-checking autofixed files...

 ✔ Formatted src/main.py
Re-checking autofixed files...


Checked 1 file
✔ No issues
Run trunk upgrade to upgrade 1 linter

trunk check --fix --filter=-ruff --print-failures src/main.py 2/2 ✓

Checked 1 file
✔ No issues

• Ran sandbox for src/api.py. ✓

Sandbox logs for https://github.com/sweepai/evals/commit/b4aaeeebfbbe7fdc5bd45c0b8693c03cdf501f9a

trunk fmt src/api.py || exit 0 1/2 ✓

 ✔ Formatted src/api.py
Re-checking autofixed files...

 ✔ Formatted src/api.py
Re-checking autofixed files...


Checked 1 file
✔ No issues
Run trunk upgrade to upgrade 1 linter

trunk check --fix --filter=-ruff --print-failures src/api.py 2/2 ✓

Checked 1 file
✔ No issues

[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.