Refactor predict function: enhance debug output by including image shape and simplifying non-zero value checks

app.py

@@ -40,12 +40,11 @@ def predict(data):
     img = np.array(data['composite'])
 
     # print non-zero values
-    print("non-zero values", np.count_nonzero(img))
+    print("non-zero values", np.count_nonzero(img), img.shape)
     for i in range(img.shape[0]):
         for j in range(img.shape[1]):
-            for k in range(img.shape[2]):
-                if img[i][j][k] != 0:
-                    print("img[i][j][k]", img[i][j][k])
+            if img[i][j] > 0:
+                print(i, j, img[i][j])
 
     print("img.shape", img.shape)