Dinoking commited on
Commit
cfaf6eb
1 Parent(s): 3a4b51c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def predict_image(img):
18
  return {class_names[i]: float(prediction[i]) for i in range(5)}
19
 
20
  image = gr.inputs.Image(shape=(180,180))
21
- label = gr.outputs.Label(num_top_classes=5)
22
  enable_queue=True
23
  description="This is a Flower Classification Model made using a CNN.Deployed to Hugging Faces using Gradio."
24
  examples = ['dandelion.jpg','sunflower.jpeg','tulip.jpg']
 
18
  return {class_names[i]: float(prediction[i]) for i in range(5)}
19
 
20
  image = gr.inputs.Image(shape=(180,180))
21
+ label = gr.outputs.Label(num_top_classes=3)
22
  enable_queue=True
23
  description="This is a Flower Classification Model made using a CNN.Deployed to Hugging Faces using Gradio."
24
  examples = ['dandelion.jpg','sunflower.jpeg','tulip.jpg']