Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -28,7 +28,7 @@ class EndpointHandler():
|
|
28 |
audio_nparray = np.frombuffer(audio_data, dtype=np.int16)
|
29 |
|
30 |
# prepare pynannote input
|
31 |
-
audio_tensor= torch.from_numpy(audio_nparray).unsqueeze(0)
|
32 |
pyannote_input = {"waveform": audio_tensor, "sample_rate": SAMPLE_RATE}
|
33 |
|
34 |
# apply pretrained pipeline
|
|
|
28 |
audio_nparray = np.frombuffer(audio_data, dtype=np.int16)
|
29 |
|
30 |
# prepare pynannote input
|
31 |
+
audio_tensor= torch.from_numpy(audio_nparray).float().unsqueeze(0)
|
32 |
pyannote_input = {"waveform": audio_tensor, "sample_rate": SAMPLE_RATE}
|
33 |
|
34 |
# apply pretrained pipeline
|