diff --git a/monstereo/predict.py b/monstereo/predict.py
index 571af47..25338cd 100644
--- a/monstereo/predict.py
+++ b/monstereo/predict.py
@@ -1,6 +1,9 @@
-
 # pylint: disable=too-many-statements, too-many-branches, undefined-loop-variable
 
+"""
+Adapted from https://github.com/vita-epfl/openpifpaf/blob/master/openpifpaf/predict.py
+"""
+
 import os
 import glob
 import json