Kresna commited on
Commit
b64f009
·
1 Parent(s): 782e3a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,10 +4,10 @@ import numpy as np
4
  from PIL import Image
5
 
6
  # Load the model
7
- model = tf.keras.models.load_model('Nutrient-Model.h5')
8
 
9
  # Define the class names
10
- class_names = ['Magnesium','Nitrogen','Potassium','Phosphorus','Sulfur']
11
 
12
  # Function to classify the image
13
  def classify_image(image):
 
4
  from PIL import Image
5
 
6
  # Load the model
7
+ model = tf.keras.models.load_model('Nutrient-Model (2).h5')
8
 
9
  # Define the class names
10
+ class_names = ['Calcium','Magnesium','Nitrogen','Phosphorus','Potassium','Sulfur']
11
 
12
  # Function to classify the image
13
  def classify_image(image):