MaxGab commited on
Commit
6f8dc49
·
verified ·
1 Parent(s): ecaacc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-smal
9
 
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
 
 
9
 
10
 
11
  def generate(description):
12
+ audio = processor(text=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