skroed commited on
Commit
97b02f0
1 Parent(s): 14d732b

fix: just return dict

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -42,4 +42,4 @@ class EndpointHandler:
42
  # postprocess the prediction
43
  prediction = outputs[0].cpu().numpy().tolist()
44
 
45
- return [{"generated_audio": prediction}]
 
42
  # postprocess the prediction
43
  prediction = outputs[0].cpu().numpy().tolist()
44
 
45
+ return {"generated_audio": prediction}