Porjaz commited on
Commit
402022e
·
verified ·
1 Parent(s): 641eeeb

Update custom_interface.py

Browse files
Files changed (1) hide show
  1. custom_interface.py +1 -1
custom_interface.py CHANGED
@@ -28,7 +28,7 @@ class ASR(Pretrained):
28
  def classify_file(self, path):
29
  # waveform = self.load_audio(path)
30
  waveform, sr = librosa.load(path, sr=16000)
31
- waveform = torch.tensor(waveform).unsqueeze(0)
32
 
33
  # Fake a batch:
34
  batch = waveform.unsqueeze(0)
 
28
  def classify_file(self, path):
29
  # waveform = self.load_audio(path)
30
  waveform, sr = librosa.load(path, sr=16000)
31
+ waveform = torch.tensor(waveform)
32
 
33
  # Fake a batch:
34
  batch = waveform.unsqueeze(0)