Update README.md

README.md

@@ -13,6 +13,7 @@ KAT model trained on ImageNet-1k (1 million images, 1,000 classes) at resolution
 KAT is a model that replaces channel mixer in transfomrers with Group Rational Kolmogorov–Arnold Network (GR-KAN).
 
 ## Usage
+The model definition is at https://github.com/Adamdad/kat, `katransformer.py`. 
 
 ```python
 from urllib.request import urlopen