Yjhhh commited on
Commit
b4d6f6d
·
verified ·
1 Parent(s): 721bd84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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