Update app.py

app.py

@@ -455,7 +455,7 @@ def process_hdf5_file(uploaded_file, percentage):
         #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}")