Update handler.py

handler.py

@@ -1,10 +1,10 @@
 from typing import Dict, List, Any
-from transformers import pipeline
+from bark import SAMPLE_RATE, generate_audio, preload_models
 
 
 class EndpointHandler:
     def __init__(self, path=""):
-        self.model = pipeline("GPT2ForSequenceClassification", "suno/bark")
+        preload_models()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
@@ -18,6 +18,10 @@ class EndpointHandler:
         text_prompt = data.pop("inputs", data)
 
         # run normal prediction
-        speech_array = self.model(text_prompt,forward_params={"do_sample": True})
-        return speech_array
+        text_prompt = """
+            Hello, my name is Suno. And, uh — and I like pizza. [laughs] 
+            But I also have other interests such as playing tic tac toe.
+        """
+        audio_array = generate_audio(text_prompt)
+        return audio_array