reach-vb HF staff commited on
Commit
0bba930
1 Parent(s): d28d084

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -33,6 +33,6 @@ class EndpointHandler:
33
  outputs = self.model.generate(**inputs,)
34
 
35
  # postprocess the prediction
36
- prediction = outputs[0].cpu().numpy()
37
 
38
  return [{"generated_audio": prediction}]
 
33
  outputs = self.model.generate(**inputs,)
34
 
35
  # postprocess the prediction
36
+ prediction = outputs[0].cpu().numpy().tolist()
37
 
38
  return [{"generated_audio": prediction}]