skroed commited on
Commit
09c5b6d
1 Parent(s): 71c2800

fix: just return dict

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