Update README.md

README.md

@@ -342,12 +342,20 @@ def main():
 if __name__ == "__main__":
     main()
 ```
-Now go forth and train, accelerate that code 🚀
-( you'll need about 22Gb of VRAM, change to batch size 1 for 8Gb) 
-Results will appear on ./capybara_finetuned_results
+🚀 Now go forth and train, accelerate that code! 
 
-Cheers, 
+> **Note:** You'll need about 22GB of VRAM. If you have 8GB, change to batch size 1.
 
-Nisten Tahiraj 
-rakun.ai 
-Toronto, Canada
+Results will appear in `./capybara_finetuned_results`
+
+---
+
+### Author
+
+**Nisten Tahiraj**  
+🏢 [rakun.ai](https://rakun.ai)  
+📍 Toronto, Canada
+
+---
+
+Happy training!