jauntybrain commited on
Commit
3408f20
1 Parent(s): 81df6e3

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -31,10 +31,10 @@ class EndpointHandler:
31
 
32
  # pass inputs with all kwargs in data
33
  with torch.autocast("cuda"):
34
- outputs = self.model.generate(**inputs, do_sample=False, max_new_tokens=400)
35
 
36
  # postprocess the prediction
37
- sampling_rate = model.config.audio_encoder.sampling_rate
38
  audio_samples = audio_values[0].cpu().numpy()[0].tolist()
39
 
40
  audio_samples = [int(min(max(sample * 32767, -32768), 32767)) for sample in audio_samples]
 
31
 
32
  # pass inputs with all kwargs in data
33
  with torch.autocast("cuda"):
34
+ audio_values = self.model.generate(**inputs, do_sample=False, max_new_tokens=400)
35
 
36
  # postprocess the prediction
37
+ sampling_rate = self.model.config.audio_encoder.sampling_rate
38
  audio_samples = audio_values[0].cpu().numpy()[0].tolist()
39
 
40
  audio_samples = [int(min(max(sample * 32767, -32768), 32767)) for sample in audio_samples]