Refactor training loop from script to class

[sweep-nightly]

Description

This PR refactors the training loop from a script to a class in order to improve code modularity and maintainability. The training loop, previously located in src/main.py, has been moved to a new class called Trainer. The Trainer class takes the model, optimizer, loss function, and training data loader as parameters and encapsulates the training process.

Summary of Changes

• Created a new class Trainer in src/main.py to encapsulate the training loop.
• Moved the training loop logic from the script to the train method of the Trainer class.
• Updated src/main.py to instantiate the Trainer class and call the train method.
• Imported the Trainer class in src/api.py to ensure compatibility with the existing code.

These changes improve code organization and make it easier to maintain and test the training process. The functionality of the existing code in src/api.py is preserved, as the trained model is still saved to the same file and loaded by the API.

Fixes #6.

---

🎉 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

[sweep-nightly]

Apply Sweep Rules to your PR?

• Apply: Leftover TODOs in the code should be handled.
• Apply: All new business logic should have corresponding unit tests in the tests/ directory.
• Apply: Any clearly inefficient or repeated code should be optimized or refactored.