Update app.py

app.py

@@ -8,10 +8,13 @@ import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
 
-# Cargar el modelo `musicgen-melody` una única vez
-model = MusicGen.get_pretrained("facebook/musicgen-melody")
 
+ZERO_GPU_PATCH_TORCH_DEVICE = 1
+
+# Cargar el modelo `musicgen-melody` una única vez
+model = MusicGen.get_pretrained("nateraw/musicgen-songstarter-v0.2")
 
+spaces.GPU()
 def generate_music(description, melody_audio, duration):
     description = clean_text(description)
     model.set_generation_params(duration=int(duration * 1000))  # Convertir segundos a milisegundos