Update app.py

app.py

@@ -5,7 +5,7 @@ from keras.models import load_model
 import numpy as np
 
 # Load the pre-trained model from the local path
-model_path = 'Leaf_namek.h5'
+model_path = 'Mango.h5'
 model = load_model(model_path)
 
 def predict_disease(image_file, model, all_labels):
@@ -39,7 +39,7 @@ def predict_disease(image_file, model, all_labels):
         return None
 
 # List of class labels
-all_labels = ['Corn-Common_rust', 'Potato-Early_blight', 'Tomato-Bacterial_spot']
+all_labels = ['Mango Anthracrose','Mango Bacterial Cancker','Mango Cutting weevil','Mango Die Back','Mango Gall Midge ','Mango Healthy','Mango powdery mildew','Mango Sooty Mould']
 
 # Define the Gradio interface
 def gradio_predict(image_file):