add missing docstrings

2024-03-24 18:59:40 +02:00 · 2024-03-24 18:59:40 +02:00 · fce5bc754f
commit fce5bc754f
parent 9b87c40b60
1 changed files with 3 additions and 0 deletions
--- 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) {