Hamza011 commited on
Commit
62a0150
1 Parent(s): 0442744

Update app.py

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