Uhhy commited on
Commit
0d75239
verified
1 Parent(s): b4d6f6d

Update app.py

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