Update app.py

app.py

@@ -10,14 +10,14 @@ model = load_model('./model.h5')
 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(1)}
+    return {class_names[i]: float(prediction[i]) for i in range(10)}
 
 
 class_names = ['Crescentia_Cujete', 'Fiddle_Wood', 'Gold_Apple', 'Hill_Mango', 'Indian_Tulip_Tree', 'Mahagony', 'Pala_Indigo_Plant', 'Spanish_Cherry', 'Teak', 'Yellow_Trumpet']
 
 image = gr.inputs.Image(shape=(224, 224))
 
-label = gr.outputs.Label(num_top_classes=1)
+label = gr.outputs.Label(num_top_classes=10)
 
 gr.Interface(fn=predict_image,
              title="Tree Classification",