reach-vb
/

musicgen-large-fp16-endpoint

Model card Files Files and versions

reach-vb HF Staff commited on Aug 2, 2023

Commit

0bba930

·

1 Parent(s): d28d084

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -33,6 +33,6 @@ class EndpointHandler:
                 outputs = self.model.generate(**inputs,)
         # postprocess the prediction
-        prediction = outputs[0].cpu().numpy()
         return [{"generated_audio": prediction}]

                 outputs = self.model.generate(**inputs,)
         # postprocess the prediction
+        prediction = outputs[0].cpu().numpy().tolist()
         return [{"generated_audio": prediction}]