Update app.py

app.py

@@ -15,13 +15,13 @@ def import_and_predict(image_data):
     x /= 255
     x=np.expand_dims(x, axis=0)
     image = np.vstack([x])
-    prediction = model.predict(image).flatten()
+    prediction = model.predict(image)[0]
     labels=class_names
     confidences = {labels[i]: float(prediction[i]) for i in range(15)}   
     return confidences
 
 gr.Interface(fn=import_and_predict, 
-             inputs=gr.Image(shape=(224, 224)),
-             outputs=gr.Label(num_top_classes=3)).launch()
+             inputs=gr.inputs.Image(shape=(224, 224)),
+             outputs=gr.outputs.Label(num_top_classes=3).launch()
 
 #examples=["banana.jpg", "car.jpg"]