KIFF commited on
Commit
a2774a3
1 Parent(s): 5775535

Update handler.py

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