Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ ZERO_GPU_PATCH_TORCH_DEVICE = 1
|
|
17 |
# Cargar el modelo `musicgen-melody` una única vez
|
18 |
model = MusicGen.get_pretrained("facebook/musicgen-melody")
|
19 |
|
|
|
20 |
def generate_music(description, melody_audio, duration):
|
21 |
description = clean_text(description)
|
22 |
model.set_generation_params(duration=int(duration * 1000)) # Convertir segundos a milisegundos
|
|
|
17 |
# Cargar el modelo `musicgen-melody` una única vez
|
18 |
model = MusicGen.get_pretrained("facebook/musicgen-melody")
|
19 |
|
20 |
+
@spaces.GPU
|
21 |
def generate_music(description, melody_audio, duration):
|
22 |
description = clean_text(description)
|
23 |
model.set_generation_params(duration=int(duration * 1000)) # Convertir segundos a milisegundos
|