MaxGab commited on
Commit
9fff38b
·
verified ·
1 Parent(s): 5e2592a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
-
4
  from transformers import AutoProcessor, MusicgenForConditionalGeneration
5
 
6
  processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
@@ -10,7 +10,8 @@ model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-smal
10
 
11
  def generate(description):
12
  audio = processor(description, padding=True, return_tensors="pt")
13
- return audio['audio']
 
14
 
15
  demo = gr.Interface(
16
  fn=generate,
 
1
  import gradio as gr
2
  from transformers import pipeline
3
+ from IPython.display import Audio
4
  from transformers import AutoProcessor, MusicgenForConditionalGeneration
5
 
6
  processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
 
10
 
11
  def generate(description):
12
  audio = processor(description, padding=True, return_tensors="pt")
13
+ audio_values = model.generate(**audio, max_new_tokens=256)
14
+ return Audio(audio_values[0].numpy(), rate=sampling_rate)
15
 
16
  demo = gr.Interface(
17
  fn=generate,