Update app.py

app.py

@@ -1,16 +1,19 @@
+import os
+import uuid
+import torch
+import re
 import spaces
 import gradio as gr
 import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
-import os
-import uuid
-import torch
-import re
 
 # Configuración del dispositivo
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
+# Configurar ZERO_GPU_PATCH_TORCH_DEVICE
+ZERO_GPU_PATCH_TORCH_DEVICE = 1
+
 # Cargar el modelo `musicgen-small` una única vez
 model = MusicGen.get_pretrained("facebook/musicgen-small")
 
@@ -63,6 +66,6 @@ gr.Interface(
     examples=[
         ["happy rock", None, 8],
         ["energetic EDM", None, 8],
-        ["chillwave", "./assets/kalhonaho.mp3", 10]
+        ["chillwave", "./assets/example_melody.mp3", 10]
     ]
 ).launch()