Update app.py

app.py

@@ -18,7 +18,7 @@ def import_and_predict(image_data):
     image = np.vstack([x])
     prediction = model.predict(image)
     labels=class_names
-    confidences = {labels[i]: float(prediction[i]) for i in range(15)}   
+    confidences = {labels[i]: float(prediction[0][i]) for i in range(15)}   
     return confidences
 
 gr.Interface(fn=import_and_predict,