GKaszewski/StableAudioWebUI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update gradio_app.py

Browse Source
This commit is contained in:
drbaph
2024-06-06 10:43:16 +01:00
committed by GitHub
parent 552dc75641
commit 2b71459752
1 changed files with 30 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
gradio_app.py
View File

@@ -10,7 +10,7 @@ from datetime import datetime
import gradio as gr import gradio as gr
# Define the function to generate audio based on a prompt # Define the function to generate audio based on a prompt
def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed): def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed, sampler_type):
device = "cuda" if torch.cuda.is_available() else "cpu" device = "cuda" if torch.cuda.is_available() else "cpu"
# Download model # Download model
@@ -36,7 +36,7 @@ def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_ti
sample_size=sample_size, sample_size=sample_size,
sigma_min=sigma_min, sigma_min=sigma_min,
sigma_max=sigma_max, sigma_max=sigma_max,
sampler_type="dpmpp-3m-sde", sampler_type=sampler_type,
device=device, device=device,
seed=seed seed=seed
) )
@@ -74,7 +74,7 @@ def generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_ti
return full_path return full_path
def audio_generator(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed): def audio_generator(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed, sampler_type):
try: try:
print("Generating audio with parameters:") print("Generating audio with parameters:")
print("Prompt:", prompt) print("Prompt:", prompt)
@@ -84,22 +84,34 @@ def audio_generator(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_t
print("Sigma Max:", sigma_max) print("Sigma Max:", sigma_max)
print("Generation Time:", generation_time) print("Generation Time:", generation_time)
print("Seed:", seed) print("Seed:", seed)
print("Sampler Type:", sampler_type)
filename = generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed) filename = generate_audio(prompt, steps, cfg_scale, sigma_min, sigma_max, generation_time, seed, sampler_type)
return gr.Audio(filename), f"Generated: {filename}" return gr.Audio(filename), f"Generated: {filename}"
except Exception as e: except Exception as e:
return str(e) return str(e)
# Create Gradio interface # Create Gradio interface
prompt_textbox = gr.Textbox(lines=5, label="Prompt") prompt_textbox = gr.Textbox(lines=5, label="Prompt")
sampler_dropdown = gr.Dropdown(
label="Sampler Type",
choices=[
"dpmpp-3m-sde",
"dpmpp-2m-sde",
"k-heun",
"k-lms",
"k-dpmpp-2s-ancestral",
"k-dpm-2",
"k-dpm-fast"
],
value="dpmpp-3m-sde"
)
steps_slider = gr.Slider(minimum=0, maximum=200, label="Steps", step=1) steps_slider = gr.Slider(minimum=0, maximum=200, label="Steps", step=1)
steps_slider.value = 100 # Set the default value here steps_slider.value = 100 # Set the default value here
cfg_scale_slider = gr.Slider(minimum=0, maximum=15, label="CFG Scale", step=1) cfg_scale_slider = gr.Slider(minimum=0, maximum=15, label="CFG Scale", step=0.1)
cfg_scale_slider.value = 7 # Set the default value here cfg_scale_slider.value = 7 # Set the default value here
sigma_min_number = gr.Number(label="Sigma Min") sigma_min_slider = gr.Slider(minimum=0, maximum=50, label="Sigma Min", step=0.1, value=0.3)
sigma_min_number.default = 0.3 # Set the default value here sigma_max_slider = gr.Slider(minimum=0, maximum=1000, label="Sigma Max", step=1, value=500)
sigma_max_number = gr.Number(label="Sigma Max")
sigma_max_number.default = 500 # Set the default value here
generation_time_slider = gr.Slider(minimum=0, maximum=47, label="Generation Time (seconds)", step=1) generation_time_slider = gr.Slider(minimum=0, maximum=47, label="Generation Time (seconds)", step=1)
generation_time_slider.value = 47 # Set the default value here generation_time_slider.value = 47 # Set the default value here
seed_slider = gr.Slider(minimum=-1, maximum=999999, label="Seed", step=1) seed_slider = gr.Slider(minimum=-1, maximum=999999, label="Seed", step=1)
@@ -107,11 +119,13 @@ seed_slider.value = 77212 # Set the default value here
output_textbox = gr.Textbox(label="Output") output_textbox = gr.Textbox(label="Output")
title = "Saganaki22 / StableAudioWebUI" title = "💀🔊 StableAudioWebUI 💀🔊"
description = "Generate audio based on a prompt. <br> (Sigma_min: 0.3, Sigma_max: 500)" description = "[Github Repository](https://github.com/Saganaki22/StableAudioWebUI)"
gr.Interface(audio_generator, gr.Interface(
[prompt_textbox, steps_slider, cfg_scale_slider, sigma_min_number, sigma_max_number, generation_time_slider, seed_slider], audio_generator,
[prompt_textbox, sampler_dropdown, steps_slider, cfg_scale_slider, sigma_min_slider, sigma_max_slider, generation_time_slider, seed_slider],
[gr.Audio(), output_textbox], [gr.Audio(), output_textbox],
title=title, title=title,
description=description).launch() description=description
).launch()