Sadjad Alikhani commited on
Commit
7a9b064
·
verified ·
1 Parent(s): dba0d85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -197,7 +197,9 @@ def process_hdf5_file(uploaded_file, percentage_idx):
197
 
198
  # Step 7: Perform inference using the functions from inference.py
199
  output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model)
 
200
  output_raw = inference.create_raw_dataset(preprocessed_chs, device)
 
201
 
202
  print(f"Output Embeddings Shape: {output_emb.shape}")
203
  print(f"Output Raw Shape: {output_raw.shape}")
 
197
 
198
  # Step 7: Perform inference using the functions from inference.py
199
  output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model)
200
+ print(f'output_emb:{output_emb[100][0]}')
201
  output_raw = inference.create_raw_dataset(preprocessed_chs, device)
202
+ print(f'output_raw:{output_raw[100][0]}')
203
 
204
  print(f"Output Embeddings Shape: {output_emb.shape}")
205
  print(f"Output Raw Shape: {output_raw.shape}")