Omnibus commited on
Commit
cfd58c5
1 Parent(s): b6974b7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoProcessor, AutoModel
3
+ import scipy
4
+
5
+
6
+ processor = AutoProcessor.from_pretrained("suno/bark-small")
7
+ model = AutoModel.from_pretrained("suno/bark-small")
8
+
9
+ def run_bark():
10
+ inputs = processor(
11
+ text=["Hello, my name is Suno. And, uh — and I like pizza. [laughs] But I also have other interests such as playing tic tac toe."],
12
+ return_tensors="pt",
13
+ )
14
+
15
+ speech_values = model.generate(**inputs, do_sample=True)
16
+
17
+ sampling_rate = model.config.sample_rate
18
+ scipy.io.wavfile.write("bark_out.wav", rate=sampling_rate, data=speech_values.cpu().numpy().squeeze())
19
+ return ("bark_out.wav")
20
+
21
+ with gr.Blocks() as app:
22
+ out_audio = gr.Audio()
23
+ go_btn = gr.Button()
24
+
25
+ go_btn.click(run_bark,None,out_audio)
26
+
27
+ app.launch()