affaaffa

app.py

@@ -77,7 +77,8 @@ def predict_image(inp):
     """
     Performs inference for a given input image and returns the prediction and CAM image.
     """
-      
+    submit_visible()
+    
     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())}
 
@@ -141,10 +142,10 @@ with gr.Blocks(title="Trash Classification", css=css) as demo:
               recycling_advice = gr.outputs.Label()
               gr.Markdown("### How to recycle?")
               recycling_qn = gr.outputs.Label()
-        def submit_visible():
-            return {results_label: gr.update(visible=True)}
+        
               
-
+    def submit_visible():
+            return {results_label: gr.update(visible=True)}
                 
     gr.Examples(
         examples=examples,
@@ -153,7 +154,7 @@ with gr.Blocks(title="Trash Classification", css=css) as demo:
         cache_examples=False,
     )
 
-    btn.click(predict_image, submit_visible, inputs=[input_image],
+    btn.click(predict_image(submit_visible()), inputs=[input_image], 
               outputs=[item_probs, recycling_qn, recycling_advice])
 if __name__ == "__main__":
     demo.launch()