Refactor prediction function: update prediction handling to print raw predictions and extract first element

app.py

@@ -63,7 +63,13 @@ def predict(data):
     img = img.reshape(1, 28, 28, 1)
 
     # Model predictions
-    preds = model.predict(img)[0]
+    preds = model.predict(img)
+
+    print(preds)
+
+    preds = preds[0]
+    print(preds)
+    
 
     # Return the probability for each class
     return {label: float(pred) for label, pred in zip(labels, preds)}