Update app.py

app.py

@@ -122,4 +122,19 @@ def predict_image(inp):
 
 with gr.Blocks(title="Trash Classification") as demo:
     gr.Markdown("# Check whether your trash is recyclable or not!", elem_id="custom_title")
-    gr.Markdown("Gradio Inference interface for classification of trash and recyclables. To use it, simply upload your image, or click one of the examples to load them. Images uploaded are never saved or indexed. No data
+    gr.Markdown("Gradio Inference interface for classification of trash and recyclables. To use it, simply upload your image, or click one of the examples to load them. Images uploaded are never saved or indexed. No data is saved. This space is free to use.", elem_id="desc1")
+    gr.Markdown("Also feel free to go through the examples shown!", elem_id="desc2")
+
+    inp = gr.Image(source="upload", type="filepath", label="Upload Your Image Here", elem_id="image_input")
+
+    material_probs = gr.Label(label="Material Classification Probabilities", elem_id="material_probs")
+    specific_probs = gr.Label(label="Specific Classification Probabilities", elem_id="specific_probs")
+    recyclable_qn = gr.Textbox(label="Item Recyclability", elem_id="recyclable_qn")
+    recyclable_advice = gr.Textbox(label="Recycling Advice", elem_id="recyclable_advice")
+
+    btn = gr.Button("Classify", elem_id="classify_btn")
+    btn.click(fn=predict_image, inputs=inp, outputs=[material_probs, specific_probs, recyclable_qn, recyclable_advice])
+
+    examples = gr.Examples(examples=examples, inputs=inp, outputs=[material_probs, specific_probs, recyclable_qn, recyclable_advice])
+
+demo.launch()