Update app.py

app.py

@@ -13,10 +13,11 @@ def maize_disease_classifier(image):
   pred,pred_idx,probs = learn_inf.predict(image)
   pred_name = f"Prediction:  {pred}"
   pred_prob = f"Probability:  {probs[pred_idx]}"
-  return pred_name, pred_prob
+  pred_array = pred_idx
+  return pred_name, pred_prob, pred_array
   
  
 iface = gr.Interface(fn=maize_disease_classifier, inputs=gr.inputs.Image(shape=(224, 224)), \
-             outputs=["text", "number"])
+             outputs=["text", "number", "text"])
 
 iface.launch(inline=False)