update some thing

Author: REBOOTERS

subs/ai/src/main/java/com/engineer/ai/util/DigitClassifier.kt

@@ -47,6 +47,7 @@ class DigitClassifier(private val context: Context) {
                 // Read input shape from model file
                 interpreter?.let { inter ->
                     val inputShape = inter.getInputTensor(0).shape()
+                    Log.d(TAG,inputShape.toString())
                     inputImageWidth = inputShape[1]
                     inputImageHeight = inputShape[2]
                     modelInputSize = FLOAT_TYPE_SIZE * inputImageWidth * inputImageHeight * PIXEL_SIZE
@@ -72,6 +73,7 @@ class DigitClassifier(private val context: Context) {
         // Post-processing: find the digit that has the highest probability
         // and return it a human-readable string.
         val result = output[0]
+        Log.d(TAG,"result = $result")
         val maxIndex = result.indices.maxBy { result[it] }
         val resultString = "Prediction Result: %d\nConfidence: %2f".format(maxIndex, result[maxIndex])
 

subs/ai/src/main/java/com/engineer/ai/util/TensorFlowLiteHelper.kt

@@ -27,8 +27,9 @@ object TensorFlowLiteHelper {
             Log.d(TAG, "ver ${TensorFlowLite.runtimeVersion(InterpreterApi.Options.TfLiteRuntime.FROM_SYSTEM_ONLY)}")
             cb(true)
         }.addOnFailureListener {
-            Log.d(TAG, "Initialized TFLite fail.")
+            Log.d(TAG, "Initialized TFLite fail")
             cb(false)
+            Log.e(TAG, "error ", it)
         }
     }