TobDeBer commited on
Commit
9e5a49b
·
1 Parent(s): 8245641

use 4s samples

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -141,7 +141,9 @@ def model_infernce(inputs):
141
  resampler = T.Resample(sample_rate, resample_rate)
142
  waveform = resampler(waveform)
143
 
144
- waveform = waveform.view(-1,) # make it (n_sample, )
 
 
145
  model_inputs = processor(waveform, sampling_rate=resample_rate, return_tensors="pt")
146
  model_inputs.to(device)
147
  with torch.no_grad():
 
141
  resampler = T.Resample(sample_rate, resample_rate)
142
  waveform = resampler(waveform)
143
 
144
+ #waveform = waveform.view(-1,) # make it (n_sample, )
145
+ waveform = waveform[0][0:4*resample_rate] # cut to 4s samples
146
+
147
  model_inputs = processor(waveform, sampling_rate=resample_rate, return_tensors="pt")
148
  model_inputs.to(device)
149
  with torch.no_grad():