Update README.md

README.md

@@ -117,6 +117,8 @@ Corrected:
 
 ![cat image, now correctly rotated](https://cdn-uploads.huggingface.co/production/uploads/697f2832c2c5e4daa93cece7/QhpbxPuwhyMj19N7v34iZ.png)
 
+*Original Image Source: [Link to Pexels](https://www.pexels.com/de-de/foto/ruhiger-schlaf-einer-getigerten-hauskatze-auf-einem-sofa-32441547/)*
+
 ## ⚠️ Limitations
 
 - Rotationally symmetric images (balls, textures, patterns) are inherently ambiguous — no model can reliably classify these.
@@ -140,6 +142,10 @@ Trained from scratch on 50k images — a pretrained backbone would likely yield
 
 > Yesterday, I was sorting photos and like every photo was rotated wrong! This inspired me to make this tool 😂
 
+## 💻 Training code
+
+The full training code can be found in `train.py`. Have fun 😊
+
 ## 📜 License
 
 Apache 2.0