Update README.md

[ashleve]

What does this PR do?

Updates README.md

Before submitting

• Did you make sure title is self-explanatory and the description concisely explains the PR?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• Did you list all the breaking changes introduced by this pull request?
• Did you test your PR locally with pytest command?
• Did you run pre-commit hooks with pre-commit run -a command?

Did you have fun?

Make sure you had fun coding 🙃

[codecov-commenter]

Codecov Report

Merging #419 (d85ac06) into main (9dbef77) will decrease coverage by 14.24%.
The diff coverage is n/a.

@@             Coverage Diff             @@
##             main     #419       +/-   ##
===========================================
- Coverage   82.90%   68.66%   -14.25%     
===========================================
  Files          11       11               
  Lines         351      351               
===========================================
- Hits          291      241       -50     
- Misses         60      110       +50     

Impacted Files	Coverage Δ
src/train.py	0.00% <0.00%> (-100.00%)	⬇️
src/utils/rich_utils.py	32.60% <0.00%> (-50.00%)	⬇️
src/utils/utils.py	50.46% <0.00%> (-14.02%)	⬇️
src/tasks/train_task.py	92.68% <0.00%> (-2.44%)	⬇️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.