ariG23498 HF staff commited on
Commit
8681243
·
1 Parent(s): 4e79da4

add: top 3

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,6 +70,6 @@ iface = gr.Interface(
70
  inputs=gr.inputs.Image(label="Input Image"),
71
  outputs=[
72
  gr.outputs.Image(label="Attention Map"),
73
- gr.outputs.Label(num_top_classes=10)
74
  ]
75
  ).launch()
 
70
  inputs=gr.inputs.Image(label="Input Image"),
71
  outputs=[
72
  gr.outputs.Image(label="Attention Map"),
73
+ gr.outputs.Label(num_top_classes=3, label="Prediction")
74
  ]
75
  ).launch()