Update gradio_app.py

Update 0.4 Check README.md
This commit is contained in:
drbaph
2024-06-10 18:02:56 +01:00
committed by GitHub
parent 34f86b8a25
commit 944e6ffce3

View File

@@ -66,7 +66,12 @@ def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_ti
save_path = os.path.join(output_folder, date_folder) save_path = os.path.join(output_folder, date_folder)
os.makedirs(save_path, exist_ok=True) os.makedirs(save_path, exist_ok=True)
# Generate a filename based on the prompt # Set a maximum filename length (e.g., 50 characters)
max_length = 50
if len(prompt) > max_length:
prompt = prompt[:max_length] + "_truncated"
# Sanitize the prompt to create a safe filename
filename = re.sub(r'\W+', '_', prompt) + ".mp3" filename = re.sub(r'\W+', '_', prompt) + ".mp3"
full_path = os.path.join(save_path, filename) full_path = os.path.join(save_path, filename)