Spaces:
Sleeping
Sleeping
saikrishna32
commited on
Commit
•
c186471
1
Parent(s):
e402601
Update app.py
Browse files
app.py
CHANGED
@@ -46,6 +46,7 @@ def get_inference(model,audio_data,audio_filepath):
|
|
46 |
waveform= waveform.T
|
47 |
model.out_layer[3]= nn.Softmax(dim=1)
|
48 |
with torch.no_grad():
|
|
|
49 |
outputs = model(waveform.to(device))
|
50 |
|
51 |
max_value, max_idx = torch.max(outputs, dim=1)
|
|
|
46 |
waveform= waveform.T
|
47 |
model.out_layer[3]= nn.Softmax(dim=1)
|
48 |
with torch.no_grad():
|
49 |
+
print(waveform.shape)
|
50 |
outputs = model(waveform.to(device))
|
51 |
|
52 |
max_value, max_idx = torch.max(outputs, dim=1)
|