Spaces:
Running
Running
Sadjad Alikhani
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -215,9 +215,9 @@ def process_hdf5_file(uploaded_file, percentage_idx):
|
|
| 215 |
|
| 216 |
# Step 7: Perform inference using the functions from inference.py
|
| 217 |
output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model)
|
| 218 |
-
print(f'output_emb:{output_emb[
|
| 219 |
output_raw = inference.create_raw_dataset(preprocessed_chs, device)
|
| 220 |
-
print(f'output_raw:{output_raw[
|
| 221 |
|
| 222 |
print(f"Output Embeddings Shape: {output_emb.shape}")
|
| 223 |
print(f"Output Raw Shape: {output_raw.shape}")
|
|
|
|
| 215 |
|
| 216 |
# Step 7: Perform inference using the functions from inference.py
|
| 217 |
output_emb = inference.lwm_inference(preprocessed_chs, 'channel_emb', model)
|
| 218 |
+
#print(f'output_emb:{output_emb[10][0]}')
|
| 219 |
output_raw = inference.create_raw_dataset(preprocessed_chs, device)
|
| 220 |
+
#print(f'output_raw:{output_raw[10][0]}')
|
| 221 |
|
| 222 |
print(f"Output Embeddings Shape: {output_emb.shape}")
|
| 223 |
print(f"Output Raw Shape: {output_raw.shape}")
|