diff --git a/src/main/kotlin/dev/nuculabs/imagetagger/ai/ImageTagsPrediction.kt b/src/main/kotlin/dev/nuculabs/imagetagger/ai/ImageTagsPrediction.kt
index 1055340..551f178 100644
--- a/src/main/kotlin/dev/nuculabs/imagetagger/ai/ImageTagsPrediction.kt
+++ b/src/main/kotlin/dev/nuculabs/imagetagger/ai/ImageTagsPrediction.kt
@@ -43,6 +43,9 @@ class ImageTagsPrediction private constructor() {
         }
     }
 
+    /**
+     * Processes an image into an ONNX Tensor.
+     */
     private fun processImage(bufferedImage: BufferedImage): Array<Array<Array<FloatArray>>> {
         try {
             val tensorData = Array(1) {