Refactor prediction function: update input key from 'image' to 'composite'

app.py

@@ -42,8 +42,8 @@ def predict(data):
     return {class_names[i]: top_3_probs[i] for i in range(3)} """
 
 def predict(data):
-    # Extract the 'image' key from the input dictionary
-    img = data['image']
+    # Extract the 'composite' key from the input dictionary
+    img = data['composite']
     # Convert to NumPy array
     img = np.array(img)