Uhhy commited on
Commit
6f03b74
1 Parent(s): 13edf2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -13,7 +13,7 @@ import torch
13
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
14
 
15
  # Cargar el modelo globalmente solo una vez
16
- logging.info("Cargando el modelo preentrenado.")
17
  model = None
18
 
19
  def cargar_modelo():
@@ -21,14 +21,15 @@ def cargar_modelo():
21
  if model is None:
22
  logging.info("Cargando MusicGen por primera vez.")
23
  model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
 
24
  else:
25
  logging.info("El modelo ya ha sido cargado anteriormente, reutilizando.")
26
 
 
 
 
27
  @spaces.GPU(duration=0)
28
  def generar_musica(descripcion, melodia_audio, duracion):
29
- # Asegurarse de que el modelo esté cargado
30
- cargar_modelo()
31
-
32
  # Limpiar la caché de la GPU antes de generar
33
  torch.cuda.empty_cache()
34
  torch.cuda.synchronize()
 
13
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
14
 
15
  # Cargar el modelo globalmente solo una vez
16
+ logging.info("Iniciando y cargando el modelo preentrenado.")
17
  model = None
18
 
19
  def cargar_modelo():
 
21
  if model is None:
22
  logging.info("Cargando MusicGen por primera vez.")
23
  model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
24
+ logging.info("Modelo cargado exitosamente.")
25
  else:
26
  logging.info("El modelo ya ha sido cargado anteriormente, reutilizando.")
27
 
28
+ # Llamar a la función para cargar el modelo inmediatamente al inicio del script
29
+ cargar_modelo()
30
+
31
  @spaces.GPU(duration=0)
32
  def generar_musica(descripcion, melodia_audio, duracion):
 
 
 
33
  # Limpiar la caché de la GPU antes de generar
34
  torch.cuda.empty_cache()
35
  torch.cuda.synchronize()