Update app.py

app.py

@@ -462,7 +462,7 @@ def process_hdf5_file(uploaded_file, percentage):
         preprocessed_chs = input_preprocess.tokenizer(manual_data=channels)
 
         # Step 7: Perform inference using the functions from inference.py
-        output_emb = inference.lwm_inference(preprocessed_chs, 'cls_emb', model, device)
+        output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model, device)
         output_raw = inference.create_raw_dataset(preprocessed_chs, device)
 
         print(f"Output Embeddings Shape: {output_emb.shape}")