Update app.py

app.py

@@ -2,12 +2,14 @@ import gradio as gr
 from keras.applications.vgg16 import VGG16
 from keras.preprocessing import image
 from keras.applications.vgg16 import preprocess_input
+from keras.applications.vgg16 import decode_predictions
 import numpy as np
 
 def predict_image(img):
     img_4d = img.reshape(-1,224,224,3)
-    prediction = model.predict(img_4d)[0]
-    return {class_names[i]: float(prediction[i]) for i in range(5)}
+    prediction = model.predict(img_4d)
+    prediction_results = decode_predictions(prediction, top = 5)
+    return { prediction_results[0][i][1]: float(prediction_results[0][i][2]) for i in range(5) }
 
 model = VGG16()
 model.summary()