Update README.md

README.md

@@ -22,8 +22,11 @@ Table of Contents:
 ## Load trained model
 ```python
 import segmentation_models_pytorch as smp
+import albumentations as A
 
-model = smp.from_pretrained("<save-directory-or-this-repo>")
+hub_repo = "commaai/comma10k-segnet"
+model = smp.from_pretrained(hub_repo)
+transform = A.Compose.from_pretrained(hub_repo)
 ```
 
 ## Model init parameters