Update app.py

app.py

@@ -214,7 +214,7 @@ def process_hdf5_file(uploaded_file, percentage_idx):
         #print(preprocessed_chs[0][0][1])
 
         # Step 7: Perform inference using the functions from inference.py
-        output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model)
+        output_emb = inference.lwm_inference(preprocessed_chs, 'cls_emb', model)
         output_raw = inference.create_raw_dataset(preprocessed_chs, device)
 
         print(f"Output Embeddings Shape: {output_emb.shape}")
@@ -301,7 +301,7 @@ with gr.Blocks(css="""
                 gr.Markdown("Percentage of Data for Training")
                 #percentage_slider_los = gr.Slider(minimum=0, maximum=4, step=1, value=0, interactive=True, elem_id="vertical-slider")
                 percentage_dropdown_los = gr.Dropdown(choices=[0, 1, 2, 3, 4, 5, 6, 7, 8],
-                                                      value=5, 
+                                                      value=0, 
                                                       label="Percentage of Data for Training", 
                                                       interactive=True)