Ignahugging commited on
Commit
f86f9d4
·
1 Parent(s): ca93fdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,11 +9,11 @@ classes=['daisy','dandelion','roses','sunflowers','tulips']
9
  def classifier(image):
10
  pred_img = modelo.predict(tf.expand_dims(image,axis=0))
11
  pred_img = tf.squeeze(tf.round(pred_img))
12
- texto = str(f'Predicted label: {classes[(np.argmax(pred_img))]}')
13
  return texto
14
 
15
  interface = gr.Interface(classifier,gr.inputs.Image(shape=(180,180)),outputs = "text",
16
- description="classifier of images of daisy plants, dandelion, roses, sunflowers, and tulips",
17
- title="Flower image classifier")
18
  interface.launch()
19
 
 
9
  def classifier(image):
10
  pred_img = modelo.predict(tf.expand_dims(image,axis=0))
11
  pred_img = tf.squeeze(tf.round(pred_img))
12
+ texto = str(f'Predicted label: {classes[(np.argmax(pred_img))]}!!!')
13
  return texto
14
 
15
  interface = gr.Interface(classifier,gr.inputs.Image(shape=(180,180)),outputs = "text",
16
+ description="Classifier of images of daisy plants, dandelion, roses, sunflowers, and tulips",
17
+ title="Flower Image Classifier")
18
  interface.launch()
19