sosss

app.py

@@ -137,8 +137,8 @@ with gr.Blocks(title="Trash Classification", css=css) as demo:
               gr.Markdown("## Outputs", elem_id="custom_header")
               gr.Markdown("### We think this is?")
               with gr.Row():
-                item_probs = gr.outputs.Label(label="Item Prediction")
-                material_probs = gr.outputs.Label(label="Material Prediction")
+                item_probs = gr.outputs.Label(label="Item Prediction",num_top_classes=1)
+                material_probs = gr.outputs.Label(label="Material Prediction", num_top_classes=1)
               gr.Markdown("### Can it Recycle?")
               recycling_qn = gr.outputs.Label()
               gr.Markdown("### How to recycle?")