amadinahmed commited on
Commit
ddfe540
·
1 Parent(s): 0dde7c3
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -122,22 +122,21 @@ def predict_image(inp):
122
  with gr.Blocks(title="Trash Classification", css="#custom_header {min-height: 3rem} #custom_title {min-height: 3rem; text-align: center}") as demo:
123
 
124
  with gr.Column():
 
 
125
  with gr.Column():
126
  gr.Markdown("## Inputs", elem_id="custom_header")
127
  input_image = gr.Image(label="Input Image")
128
  input_image.style(height=240)
129
  btn = gr.Button(value="Submit")
130
  btn.style(full_width=True)
131
- with gr.Column(variant="panel", scale=2):
132
  gr.Markdown("## Outputs", elem_id="custom_header")
133
  gr.Markdown("### Can it Recycle?")
134
  recycling_qn = gr.outputs.Label()
135
  gr.Markdown("### How to recycle?")
136
  recycling_advice = gr.outputs.Label()
137
- item_probs = gr.outputs.Label()
138
- with gr.Row():
139
- material_probs = gr.outputs.Label()
140
- item_probs = gr.outputs.Label()
141
 
142
  gr.Examples(
143
  examples=examples,
 
122
  with gr.Blocks(title="Trash Classification", css="#custom_header {min-height: 3rem} #custom_title {min-height: 3rem; text-align: center}") as demo:
123
 
124
  with gr.Column():
125
+ with gr.Row():
126
+ item_probs = gr.outputs.Label()
127
  with gr.Column():
128
  gr.Markdown("## Inputs", elem_id="custom_header")
129
  input_image = gr.Image(label="Input Image")
130
  input_image.style(height=240)
131
  btn = gr.Button(value="Submit")
132
  btn.style(full_width=True)
133
+ with gr.Column():
134
  gr.Markdown("## Outputs", elem_id="custom_header")
135
  gr.Markdown("### Can it Recycle?")
136
  recycling_qn = gr.outputs.Label()
137
  gr.Markdown("### How to recycle?")
138
  recycling_advice = gr.outputs.Label()
139
+
 
 
 
140
 
141
  gr.Examples(
142
  examples=examples,