Update app.py

app.py

@@ -3,7 +3,6 @@ import gradio as gr
 import torchaudio
 from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
-import logging
 import os
 import uuid
 from torch.cuda.amp import autocast
@@ -12,54 +11,35 @@ import re
 
 ZERO_GPU_PATCH_TORCH_DEVICE = 1
 
-# Configura o logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
-
-logging.info("Carregando o modelo pré-treinado.")
 model = MusicGen.get_pretrained('nateraw/musicgen-songstarter-v0.2')
 
 
 @spaces.GPU(duration=120)
 def generate_music(description, melody_audio, duration):
-    # Remove spam da descrição
     description = clean_text(description)
-
-    # Define a duração da música em segundos
     model.set_generation_params(duration=int(duration * 1000))
 
     with autocast():
-        logging.info("Iniciando a geração de música.")
         if description:
             description = [description]
             if melody_audio:
-                logging.info(f"Carregando a melodia de áudio de: {melody_audio}")
                 melody, sr = torchaudio.load(melody_audio)
-                logging.info("Gerando música com descrição e melodia.")
                 wav = model.generate_with_chroma(description, melody[None], sr)
             else:
-                logging.info("Gerando música apenas com descrição.")
                 wav = model.generate(description)
         else:
-            logging.info("Gerando música de forma incondicional.")
             wav = model.generate_unconditional(1)
         filename = f'{str(uuid.uuid4())}.wav'
-        logging.info(f"Salvando a música gerada com o nome: {filename}")
         path = audio_write(filename, wav[0].cpu().to(torch.float32), model.sample_rate, strategy="loudness", loudness_compressor=True)
-        print("Música salva em", path, ".")
-        logging.info(f"A forma do tensor de áudio gerado: {wav[0].shape}")
-        logging.info("Música gerada e salva com sucesso.")
         if not os.path.exists(path):
             raise ValueError(f'Failed to save audio to {path}')
-
         return path
 
 def clean_text(text):
-    """Remove links e caracteres especiais de um texto."""
-    text = re.sub(r'http\S+', '', text)  # remove links
-    text = re.sub(r'[^a-zA-Z0-9\s]', '', text)  # remove caracteres especiais
+    text = re.sub(r'http\S+', '', text)
+    text = re.sub(r'[^a-zA-Z0-9\s]', '', text)
     return text
 
-# Define a interface Gradio
 description = gr.Textbox(label="Descrição", placeholder="acústico, guitarra, melodia, trap, ré menor, 90 bpm")
 melody_audio = gr.Audio(label="Melodia de Áudio (opcional)", type="filepath")
 duration = gr.Number(label="Duração (segundos)", value=10)