lukelike1001 commited on
Commit
f5640b9
1 Parent(s): 1f32fd0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from tensorflow import keras
6
 
7
  # load the pre-trained model from the appropriate file path
8
  def predict_plant(path):
9
- model = tf.keras.models.load_model('my_model/')
10
 
11
  # redefine values from the model
12
  img_height = img_width = 180
 
6
 
7
  # load the pre-trained model from the appropriate file path
8
  def predict_plant(path):
9
+ model = tf.saved_model.load('my_model/')
10
 
11
  # redefine values from the model
12
  img_height = img_width = 180