Update handler.py

handler.py

@@ -8,7 +8,7 @@ class EndpointHandler:
         self.processor = AutoProcessor.from_pretrained(path)
         self.model = MusicgenForConditionalGeneration.from_pretrained(path, torch_dtype=torch.float16).to("cuda")
 
-    def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
+    def __call__(self, data: Dict[str, Any]) -> bytes:
         """
         Args:
             data (:dict:):