Samuel L Meyers
commited on
Commit
·
8688550
1
Parent(s):
96b4603
Hopefully fixed app, even if it's slow.
Browse files
app.py
CHANGED
@@ -4,15 +4,14 @@ import scipy
|
|
4 |
import torch
|
5 |
|
6 |
processor = AutoProcessor.from_pretrained("suno/bark-small")
|
7 |
-
model = BarkModel.from_pretrained("suno/bark-small"
|
8 |
-
model = model.to_bettertransformer()
|
9 |
|
10 |
def greet(text):
|
11 |
inputs = processor(
|
12 |
text=[text],
|
13 |
return_tensors="pt",
|
14 |
)
|
15 |
-
speech_values = model.generate(**inputs, do_sample=True)
|
16 |
scipy.io.wavfile.write("tmp.wav", rate=24000, data=speech_values.cpu().numpy().squeeze())
|
17 |
return open("tmp.wav", "rb").read()
|
18 |
|
|
|
4 |
import torch
|
5 |
|
6 |
processor = AutoProcessor.from_pretrained("suno/bark-small")
|
7 |
+
model = BarkModel.from_pretrained("suno/bark-small")
|
|
|
8 |
|
9 |
def greet(text):
|
10 |
inputs = processor(
|
11 |
text=[text],
|
12 |
return_tensors="pt",
|
13 |
)
|
14 |
+
speech_values = model.generate(**inputs, do_sample=True, num_beams=5)
|
15 |
scipy.io.wavfile.write("tmp.wav", rate=24000, data=speech_values.cpu().numpy().squeeze())
|
16 |
return open("tmp.wav", "rb").read()
|
17 |
|