Update app.py

app.py

@@ -85,8 +85,8 @@ def sepia(img,plant_type):
             print(disease,y_pred[0][disease],max_prob_indx,y_pred[0][max_prob_indx])
             if y_pred[0][disease]>y_pred[0][max_prob_indx]:
                 max_prob_indx = disease
-    print(label_disease[indx],y_pred[indx],label_disease[max_prob_indx],y_pred[max_prob_indx])
-    return label_disease[indx],y_pred[indx],label_disease[max_prob_indx],y_pred[max_prob_indx]
+    print(label_disease[indx],y_pred[0][indx],label_disease[max_prob_indx],y_pred[0][max_prob_indx])
+    return label_disease[indx],y_pred[0][indx],label_disease[max_prob_indx],y_pred[0][max_prob_indx]
 
 demo = gr.Interface(sepia, [gr.Image(),gr.Radio(["Apple","Blueberry","Cherry","Corn","Grape","Orange","Peach","Pepper","Potato","Raspberry","Soybean","Squash","Strawberry","Tomato"])],["text","number","text","number"])
 demo.launch(share=True)