Master0fNone commited on
Commit
c9a4866
·
verified ·
1 Parent(s): 8fb3c7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+
4
+ synthesiser = pipeline("text-to-audio", model="facebook/musicgen-small")
5
+ def text_to_audio(prompt):
6
+ music = synthesiser(prompt, forward_params={"do_sample": True})
7
+ output_audio = music['audio']
8
+ sampling_rate = music['sampling_rate']
9
+ return sampling_rate, output_audio
10
+ demo = gr.Interface(
11
+ fn=text_to_audio,
12
+ inputs=gr.Textbox(label="Music Prompt", placeholder="Enter text for audio gen. Ex: A calm piano melody."),
13
+ outputs=gr.Audio(label="Generated Audio"),
14
+ title="MusicGen Audio Test",
15
+ description="Generate audio based on text prompts using a pretrained model",
16
+ )
17
+ demo.launch()