Spaces:
Running
Running
Sadjad Alikhani
commited on
Update app.py
Browse files
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}")
|