MaxGab commited on
Commit
64e0350
·
verified ·
1 Parent(s): aff0444

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -5
app.py CHANGED
@@ -1,6 +1,16 @@
1
- import gradio as gr
2
 
3
- gr.load(
4
- "models/facebook/musicgen-small",
5
- provider="hf-inference",
6
- ).launch()
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
 
3
+ pipe = pipeline("text-to-music", model="openai/jukebox")
4
+
5
+ def generate(description):
6
+ audio = pipe(description)
7
+ return audio['audio']
8
+
9
+ demo = gr.Interface(
10
+ fn=generate,
11
+ inputs=gr.Textbox(label="Enter Text to Convert to Audio"),
12
+ outputs=gr.Audio(label="Generated Audio"),
13
+ live=True
14
+ )
15
+
16
+ demo.launch()