Refactor predict function: streamline image extraction and add debug prints for image shape and content

app.py

@@ -36,10 +36,10 @@ labels = {
 model = tf.keras.models.load_model("./sketch_recognition_numbers_model.h5")
 
 def predict(data):
-    # Extract the 'composite' key from the input dictionary
-    img = data['composite']
     # Convert to NumPy array
-    img = np.array(img)
+    img = np.array(data['composite'])
+
+    print("img.shape", img.shape)
 
     # Handle RGBA or RGB images
     if img.shape[-1] == 4:  # RGBA
@@ -56,6 +56,8 @@ def predict(data):
     # Reshape to match model input
     img = img.reshape(1, 28, 28, 1)
 
+    print("img", img)
+
     # Model predictions
     preds = model.predict(img)[0]