Update app.py

app.py

@@ -43,7 +43,7 @@ def predict(image):
     data[0] = normalized_image_array
 
     # run the inference
-    prediction = model.predict(data)    
+    prediction = model.predict(data)[0]
     confidences = {label_dict[i]: float(prediction[i]) for i in range(4)}    
     
     return confidences