Update handler.py

handler.py

@@ -1,10 +1,10 @@
 from typing import Dict, List, Any
-from bark import SAMPLE_RATE, generate_audio, preload_models
+from transformers import pipeline
 
 
 class EndpointHandler:
     def __init__(self, path=""):
-        preload_models()
+        self.model = pipeline("text-to-speech", "suno/bark")
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
@@ -18,5 +18,6 @@ class EndpointHandler:
         text_prompt = data.pop("inputs", data)
 
         # run normal prediction
-        speech_array = generate_audio(text_prompt)
+        speech_array = self.model(text_prompt,forward_params={"do_sample": True})
         return speech_array
+