saikrishna32 commited on
Commit
c186471
1 Parent(s): e402601

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)