Update app.py

app.py

@@ -4,11 +4,10 @@ import numpy as np
 from PIL import Image
 
 # Load the model
-model = tf.keras.models.load_model('Nutrient-Model.h5')
+model = tf.keras.models.load_model('Adam_8_1000_Acc 0.88_Nutrient-Model.h5')
 
 # Define the class names
-class_names = ['Magnesium','Nitrogen','Potassium','Phosphorus','Sulfur']
-
+class_names = ['Calcium','Magnesium','Nitrogen','Phosphorus','Potassium','Sulfur']
 # Function to classify the image
 def classify_image(image):
     # Convert the numpy array to a PIL Image object