Update app.py

app.py

@@ -2,7 +2,10 @@ import gradio as gr
 from datasets import load_dataset
 
 # --- Configuration ---
-DATASET_NAME = "Cnam-LMSSC/vibravox"
+# The ONLY change is on this line: we're pointing back to the smaller test dataset.
+DATASET_NAME = "Cnam-LMSSC/vibravox-test"
+# ---------------------------------------------------------------------------------
+
 SUBSETS = ["speech_clean", "speech_noisy", "speechless_clean", "speechless_noisy"]
 SPLITS = ["train", "validation", "test"]
 TEXT_COLUMN = "raw_text"
@@ -81,10 +84,9 @@ def get_audio_row(dataset, index):
 
 # --- Build the Gradio Interface ---
 with gr.Blocks(css="footer {display: none !important}") as demo:
-    gr.Markdown("# Vibravox Multi-Sensor Explorer")
+    gr.Markdown("# Vibravox Multi-Sensor Explorer (Test Dataset)")
 
     # This state object holds the currently loaded dataset in memory
-    # It's invisible to the user but accessible by our functions
     loaded_dataset_state = gr.State(None)
 
     # --- INPUT CONTROLS ---
@@ -106,20 +108,15 @@ with gr.Blocks(css="footer {display: none !important}") as demo:
         audio5 = gr.Audio(label="Forehead Accelerometer")
         audio6 = gr.Audio(label="Temple Vibration Pickup")
 
-    # A list of all the output components for easier reference
     all_outputs = [loaded_dataset_state, slider, sentence_output, audio1, audio2, audio3, audio4, audio5, audio6, error_box]
     audio_outputs = [sentence_output, audio1, audio2, audio3, audio4, audio5, audio6]
 
     # --- WIRING THE EVENT HANDLERS ---
-    
-    # 1. When the app first loads, run the main function with default values
     demo.load(
         fn=load_and_update_all,
         inputs=[subset_dropdown, split_dropdown],
         outputs=all_outputs
     )
-    
-    # 2. When a dropdown value changes, re-run the main function
     subset_dropdown.change(
         fn=load_and_update_all,
         inputs=[subset_dropdown, split_dropdown],
@@ -130,8 +127,6 @@ with gr.Blocks(css="footer {display: none !important}") as demo:
         inputs=[subset_dropdown, split_dropdown],
         outputs=all_outputs
     )
-
-    # 3. When ONLY the slider changes, run the simpler function
     slider.change(
         fn=get_audio_row,
         inputs=[loaded_dataset_state, slider],