Update README.md

README.md

@@ -25,4 +25,10 @@ This is the pre-trained PyTorch model for **NextClick**, a sequential recommenda
 import torch
 # Load Architecture code first (See GitHub), then:
 model.load_state_dict(torch.load("sasrec_model.pth"))
-model.eval()
+model.eval()
+```
+
+### GitHub Repository
+For the training code and architecture definition, visit:
+[https://github.com/eatosin/NextClick-RecSys-Engine](https://github.com/eatosin/NextClick-RecSys-Engine)
+```