Update main.py

[alikhan1998]

Details
The provided script is a part of the OncoNet project and is used for training, evaluating, and predicting on a dataset. While the script appears to be functional, there are opportunities for improvement in terms of code organization and documentation.

Changes

1. Modularize the Code: Break down the script into smaller, well-defined functions or classes to improve code structure and maintainability.

2. Add Comments and Documentation: Document complex logic or provide context for certain operations to enhance code understanding.

3. Improve Variable Naming: Ensure that variable and function names are descriptive and follow a consistent naming convention.

4. Implement Error Handling: Add appropriate error handling to handle scenarios where files or resources may not be found.

5. Utilize os.path.join(): Use this function to construct file paths, ensuring compatibility across different operating systems