File size: 669 Bytes
3e55f57 62f977c 3e55f57 f067669 3e55f57 62f977c 3e55f57 f067669 3e55f57 f067669 6eea13c f067669 62f977c 072ee69 62f977c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from typing import Dict, List, Any
from transformers import pipeline
class EndpointHandler:
def __init__(self, path=""):
self.model = pipeline("text-to-speech", "suno/bark")
def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
"""
data args:
inputs (:obj: `str`)
date (:obj: `str`)
Return:
A :obj:`list` | `dict`: will be serialized and returned
"""
# get inputs
text_prompt = data.pop("inputs", data)
# run normal prediction
speech_array = self.model(text_prompt,forward_params={"do_sample": True})
return speech_array
|