adaadda

app.py

@@ -77,8 +77,6 @@ def predict_image(inp):
     """
     Performs inference for a given input image and returns the prediction and CAM image.
     """
-
-    
     material_label, material_label_idx, material_probs = materials_model.predict(inp)
     material_preds = {name: prob for name, prob in zip(material_names, material_probs.tolist())}
 
@@ -119,6 +117,7 @@ def predict_image(inp):
 
     return material_preds, specific_preds, recyclable_qn, recyclable_advice
 
+
 css = """
 #warning {background-color: #FFCCCB}
 #custom_header {min-height: 3rem} #custom_title {min-height: 3rem; text-align: center}
@@ -141,10 +140,12 @@ with gr.Blocks(title="Trash Classification", css=css) as demo:
               gr.Markdown("### Can it Recycle?")
               recycling_advice = gr.outputs.Label()
               gr.Markdown("### How to recycle?")
-              item_probs = gr.outputs.Label(label="Item Prediction")
+              with gr.Row():
+                material_probs = gr.outputs.Label(label="Material Prediction")
+                item_probs = gr.outputs.Label(label="Item Prediction")
                 
 
-    btn.click(predict_image, inputs=[input_image], 
-              outputs=[item_probs, recycling_qn, recycling_advice])
+    btn.click(predict_image, inputs=[input_image],
+              outputs=[material_probs, item_probs, recycling_qn, recycling_advice])
 if __name__ == "__main__":
     demo.launch()