amadinahmed commited on
Commit
f88d5c1
·
1 Parent(s): 2f0905b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -127,7 +127,7 @@ with gr.Blocks(title="Trash Classification", css=css) as demo:
127
 
128
  with gr.Column():
129
  with gr.Row():
130
- item_probs = gr.outputs.Label(num_top_classes=0)
131
  with gr.Column():
132
  gr.Markdown("## Inputs", elem_id="custom_header")
133
  input_image = gr.Image(label="Input Image")
 
127
 
128
  with gr.Column():
129
  with gr.Row():
130
+ item_probs = gr.outputs.Label(num_top_classes=1)
131
  with gr.Column():
132
  gr.Markdown("## Inputs", elem_id="custom_header")
133
  input_image = gr.Image(label="Input Image")