malihoseini commited on
Commit
0e71c42
·
verified ·
1 Parent(s): 740f27a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ mobile_net = tf.keras.applications.MobileNetV2()
12
  def image_classifier(img):
13
  arr = np.expand_dims(img, axis=0)
14
  arr = tf.keras.applications.mobilenet.preprocess_input(arr)
15
- prediction = mobile_net.prediction(arr).flatten()
16
  return {labels[i]:float(prediction[i]) for i in range(1000)}
17
  iface = gr.Interface(
18
  image_classifier,
 
12
  def image_classifier(img):
13
  arr = np.expand_dims(img, axis=0)
14
  arr = tf.keras.applications.mobilenet.preprocess_input(arr)
15
+ prediction = mobile_net.predict(arr).flatten()
16
  return {labels[i]:float(prediction[i]) for i in range(1000)}
17
  iface = gr.Interface(
18
  image_classifier,