Update app.py

app.py

@@ -94,8 +94,11 @@ def classify_toxicity(audio_file, selected_sounds, viz_state, slider):
             if score > threshold:
                 print(f"Threshold exceeded for class '{selected_class_name}': Score = {score:.4f}")
                 viz_state = True
+                calm_image : gr.update(visible=True)
             else:
                 viz_state = False
+                calm_image : gr.update(visible=False)
+
 
 
     '''
@@ -130,7 +133,7 @@ with gr.Blocks() as iface:
     with gr.Column():
         # out_val = gr.Textbox()
         out_class = gr.Label()
-        calm_image = gr.Image(value="./hrv-breathing.gif", visible=show_state)
+        calm_image = gr.Image(value="./hrv-breathing.gif", visible=False)
     submit_btn.click(fn=classify_toxicity, inputs=[aud_input, miso_sounds, show_state, sense_slider], outputs=[out_class, calm_image])