Update custom_interface.py
Browse files- 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)
|
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)
|