Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,16 +8,10 @@ import torchaudio
|
|
8 |
from audiocraft.models import MusicGen
|
9 |
from audiocraft.data.audio import audio_write
|
10 |
|
11 |
-
# Configuraci贸n del dispositivo
|
12 |
-
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
13 |
-
|
14 |
-
# Configurar ZERO_GPU_PATCH_TORCH_DEVICE
|
15 |
-
ZERO_GPU_PATCH_TORCH_DEVICE = 1
|
16 |
-
|
17 |
# Cargar el modelo `musicgen-melody` una 煤nica vez
|
18 |
model = MusicGen.get_pretrained("facebook/musicgen-melody")
|
19 |
|
20 |
-
|
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
|
|
|
8 |
from audiocraft.models import MusicGen
|
9 |
from audiocraft.data.audio import audio_write
|
10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
# Cargar el modelo `musicgen-melody` una 煤nica vez
|
12 |
model = MusicGen.get_pretrained("facebook/musicgen-melody")
|
13 |
|
14 |
+
|
15 |
def generate_music(description, melody_audio, duration):
|
16 |
description = clean_text(description)
|
17 |
model.set_generation_params(duration=int(duration * 1000)) # Convertir segundos a milisegundos
|