Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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("
|
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()
|