Update app.py

app.py

@@ -12,11 +12,19 @@ import torch
 # Configura el logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# Limitar el uso de memoria de la GPU
+torch.cuda.set_per_process_memory_fraction(0.8)
+
+# Cargar el modelo preentrenado
 logging.info("Cargando el modelo preentrenado.")
 model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
 
-@spaces.GPU(duration=0)
+@spaces.GPU(duration=120)
 def generar_musica(descripcion, melodia_audio, duracion):
+    # Limpiar la caché de la GPU antes de generar
+    torch.cuda.empty_cache()
+    torch.cuda.synchronize()
+
     with autocast():
         logging.info("Iniciando la generación de música.")
         model.set_generation_params(duration=duracion)