diff --git a/tracking_re_id/tracking_re_id/KeyRe_ID_model.py b/tracking_re_id/tracking_re_id/KeyRe_ID_model.py
index 1aeb759..87f4f0e 100644
--- a/tracking_re_id/tracking_re_id/KeyRe_ID_model.py
+++ b/tracking_re_id/tracking_re_id/KeyRe_ID_model.py
@@ -1,3 +1,5 @@
+# Source: https://github.com/jinseong0115/KeyRe-ID
+
 import torch
 import torch.nn as nn
 import copy