Update app.py

app.py

@@ -335,8 +335,8 @@ with gr.Blocks(css="""
             embeddings_img_bp = gr.Image(label="Embeddings", type="pil", width=300, height=300)
 
         # Update the confusion matrices whenever sliders change
-        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_slider], outputs=[raw_img_bp, embeddings_img_bp])
-        task_complexity_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_slider], outputs=[raw_img_bp, embeddings_img_bp])
+        data_percentage_slider.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
+        task_complexity_dropdown.change(fn=beam_prediction_task, inputs=[data_percentage_slider, task_complexity_dropdown], outputs=[raw_img_bp, embeddings_img_bp])
 
     # Separate Tab for LoS/NLoS Classification Task
     with gr.Tab("LoS/NLoS Classification Task"):