malihoseini commited on
Commit
3df1d3f
·
verified ·
1 Parent(s): 1f55972

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ current_dir = dirname(realpath(__file__))
9
  with open(join(current_dir, 'image_labels.json')) as labels_file:
10
  labels=json.load(labels_file)
11
 
12
- mobile_net = tf.keras.applications.MobileNetV3Large()
13
  def image_classifier(img):
14
  img = cv2.resize(img, (224,224))
15
  arr = np.expand_dims(img, axis=0)
 
9
  with open(join(current_dir, 'image_labels.json')) as labels_file:
10
  labels=json.load(labels_file)
11
 
12
+ mobile_net = tf.keras.applications.MobileNetV2()
13
  def image_classifier(img):
14
  img = cv2.resize(img, (224,224))
15
  arr = np.expand_dims(img, axis=0)