Color format the code snippet in model card

Update readme file for color formatting the code snippet

README.md

@@ -24,7 +24,7 @@ Training recipe:
 
 - GitHub: https://github.com/apple/ml-diffucoder
 
-```
+```py
 import torch
 from transformers import AutoModel, AutoTokenizer