From fce5bc754f7162cf57a7db52d8eba095069b0ec2 Mon Sep 17 00:00:00 2001
From: Denis Nutiu <denis.nutiu@gmail.com>
Date: Sun, 24 Mar 2024 18:59:40 +0200
Subject: [PATCH] add missing docstrings

---
 .../kotlin/dev/nuculabs/imagetagger/ai/ImageTagsPrediction.kt  | 3 +++
 1 file changed, 3 insertions(+)

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) {