initial commit
This commit is contained in:
12
t2a_api.py
Normal file
12
t2a_api.py
Normal file
@@ -0,0 +1,12 @@
|
||||
from transformers import pipeline
|
||||
import scipy
|
||||
|
||||
|
||||
class T2A(object):
|
||||
def __init__(self):
|
||||
self.synthesiser = pipeline("text-to-speech", "suno/bark")
|
||||
|
||||
def apply(self, text):
|
||||
speech = self.synthesiser(text,
|
||||
forward_params={"do_sample": True})
|
||||
scipy.io.wavfile.write("bark_out.wav", rate=speech["sampling_rate"], data=speech["audio"])
|
||||
Reference in New Issue
Block a user