Spaces:
Running
Running
Update app.py
Browse files
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 |
|