HashamUllah commited on
Commit
b950a55
1 Parent(s): 69f9410

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import json
7
 
8
  app = Flask(__name__)
9
 
10
- # Load the TensorFlow model
11
- model = tf.keras.models.load_model('plant_disease_detection_saved_model')
12
 
13
  # Load categories
14
  with open('categories.json') as f:
 
7
 
8
  app = Flask(__name__)
9
 
10
+ # Load the H5 model
11
+ model = tf.keras.models.load_model('plant_disease_detection.h5')
12
 
13
  # Load categories
14
  with open('categories.json') as f: