Hugging Face's logo

Spaces:
Hamza011
/
maize-classifier
Sleeping

App Files Community
Hamza011 commited on
Commit
aaa5af9
·
verified ·
1 Parent(s): dbd53d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
 
6
 
7
  print(os.listdir('./model'))
8
- used_model = tf.keras.layers.TFSMLayer('./model')
9
  new_classes = ['blight', 'common_rust', 'gray_leaf_spot','healthy']
10
 
11
  def classify_image(img_dt):
 
5
 
6
 
7
  print(os.listdir('./model'))
8
+ used_model = tf.keras.models.load_model ('./model')
9
  new_classes = ['blight', 'common_rust', 'gray_leaf_spot','healthy']
10
 
11
  def classify_image(img_dt):