Upload 2 files

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,10 +18,6 @@ class EndpointHandler:
         text_prompt = data.pop("inputs", data)
 
         # run normal prediction
-        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
+        speech_array = self.model(text_prompt,forward_params={"do_sample": True})
+        return speech_array
 

requirements.txt

@@ -1 +1 @@
-bark
+transformers