Spaces:
Paused
Paused
fix default values
Browse files
stable_audio_tools/interface/gradio.py
CHANGED
@@ -373,7 +373,7 @@ def create_uncond_sampling_ui(model_config):
|
|
373 |
audio_spectrogram_output
|
374 |
],
|
375 |
api_name="generate")
|
376 |
-
def create_conditioning_slider(min_val, max_val, label):
|
377 |
"""
|
378 |
Create a Gradio slider for a given conditioning parameter.
|
379 |
|
@@ -386,7 +386,7 @@ def create_conditioning_slider(min_val, max_val, label):
|
|
386 |
- A gr.Slider object configured according to the provided parameters.
|
387 |
"""
|
388 |
step = (max_val - min_val) / 1000
|
389 |
-
default_val =
|
390 |
print(f"Creating slider for {label} with min_val={min_val}, max_val={max_val}, step={step}, default_val={default_val}")
|
391 |
return gr.Slider(minimum=min_val, maximum=max_val, step=step, value=default_val, label=label)
|
392 |
|
@@ -431,6 +431,7 @@ def create_sampling_ui(model_config):
|
|
431 |
latitude_slider = create_conditioning_slider(
|
432 |
min_val=latitude_config["config"]["min_val"],
|
433 |
max_val=latitude_config["config"]["max_val"],
|
|
|
434 |
label="latitude")
|
435 |
|
436 |
longitude_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "longitude"), None)
|
@@ -438,6 +439,7 @@ def create_sampling_ui(model_config):
|
|
438 |
longitude_slider = create_conditioning_slider(
|
439 |
min_val=longitude_config["config"]["min_val"],
|
440 |
max_val=longitude_config["config"]["max_val"],
|
|
|
441 |
label="longitude")
|
442 |
|
443 |
temperature_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "temperature"), None)
|
@@ -445,6 +447,7 @@ def create_sampling_ui(model_config):
|
|
445 |
temperature_slider = create_conditioning_slider(
|
446 |
min_val=temperature_config["config"]["min_val"],
|
447 |
max_val=temperature_config["config"]["max_val"],
|
|
|
448 |
label="temperature")
|
449 |
|
450 |
humidity_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "humidity"), None)
|
@@ -452,6 +455,7 @@ def create_sampling_ui(model_config):
|
|
452 |
humidity_slider = create_conditioning_slider(
|
453 |
min_val=humidity_config["config"]["min_val"],
|
454 |
max_val=humidity_config["config"]["max_val"],
|
|
|
455 |
label="humidity")
|
456 |
|
457 |
wind_speed_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "wind_speed"), None)
|
@@ -459,6 +463,7 @@ def create_sampling_ui(model_config):
|
|
459 |
wind_speed_slider = create_conditioning_slider(
|
460 |
min_val=wind_speed_config["config"]["min_val"],
|
461 |
max_val=wind_speed_config["config"]["max_val"],
|
|
|
462 |
label="wind_speed")
|
463 |
|
464 |
pressure_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "pressure"), None)
|
@@ -466,6 +471,7 @@ def create_sampling_ui(model_config):
|
|
466 |
pressure_slider = create_conditioning_slider(
|
467 |
min_val=pressure_config["config"]["min_val"],
|
468 |
max_val=pressure_config["config"]["max_val"],
|
|
|
469 |
label="pressure")
|
470 |
|
471 |
minutes_of_day_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "minutes_of_day"), None)
|
@@ -473,6 +479,7 @@ def create_sampling_ui(model_config):
|
|
473 |
minutes_of_day_slider = create_conditioning_slider(
|
474 |
min_val=minutes_of_day_config["config"]["min_val"],
|
475 |
max_val=minutes_of_day_config["config"]["max_val"],
|
|
|
476 |
label="minutes_of_day")
|
477 |
|
478 |
day_of_year_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "day_of_year"), None)
|
@@ -480,6 +487,7 @@ def create_sampling_ui(model_config):
|
|
480 |
day_of_year_slider = create_conditioning_slider(
|
481 |
min_val=day_of_year_config["config"]["min_val"],
|
482 |
max_val=day_of_year_config["config"]["max_val"],
|
|
|
483 |
label="Day of year")
|
484 |
|
485 |
with gr.Accordion("Sampler params", open=False):
|
|
|
373 |
audio_spectrogram_output
|
374 |
],
|
375 |
api_name="generate")
|
376 |
+
def create_conditioning_slider(min_val, max_val, default_value, label):
|
377 |
"""
|
378 |
Create a Gradio slider for a given conditioning parameter.
|
379 |
|
|
|
386 |
- A gr.Slider object configured according to the provided parameters.
|
387 |
"""
|
388 |
step = (max_val - min_val) / 1000
|
389 |
+
default_val = default_value
|
390 |
print(f"Creating slider for {label} with min_val={min_val}, max_val={max_val}, step={step}, default_val={default_val}")
|
391 |
return gr.Slider(minimum=min_val, maximum=max_val, step=step, value=default_val, label=label)
|
392 |
|
|
|
431 |
latitude_slider = create_conditioning_slider(
|
432 |
min_val=latitude_config["config"]["min_val"],
|
433 |
max_val=latitude_config["config"]["max_val"],
|
434 |
+
default_value = -26.9773,
|
435 |
label="latitude")
|
436 |
|
437 |
longitude_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "longitude"), None)
|
|
|
439 |
longitude_slider = create_conditioning_slider(
|
440 |
min_val=longitude_config["config"]["min_val"],
|
441 |
max_val=longitude_config["config"]["max_val"],
|
442 |
+
default_value=152.1714,
|
443 |
label="longitude")
|
444 |
|
445 |
temperature_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "temperature"), None)
|
|
|
447 |
temperature_slider = create_conditioning_slider(
|
448 |
min_val=temperature_config["config"]["min_val"],
|
449 |
max_val=temperature_config["config"]["max_val"],
|
450 |
+
default_value=12.93,
|
451 |
label="temperature")
|
452 |
|
453 |
humidity_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "humidity"), None)
|
|
|
455 |
humidity_slider = create_conditioning_slider(
|
456 |
min_val=humidity_config["config"]["min_val"],
|
457 |
max_val=humidity_config["config"]["max_val"],
|
458 |
+
default_value=98,
|
459 |
label="humidity")
|
460 |
|
461 |
wind_speed_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "wind_speed"), None)
|
|
|
463 |
wind_speed_slider = create_conditioning_slider(
|
464 |
min_val=wind_speed_config["config"]["min_val"],
|
465 |
max_val=wind_speed_config["config"]["max_val"],
|
466 |
+
default_value=1.66,
|
467 |
label="wind_speed")
|
468 |
|
469 |
pressure_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "pressure"), None)
|
|
|
471 |
pressure_slider = create_conditioning_slider(
|
472 |
min_val=pressure_config["config"]["min_val"],
|
473 |
max_val=pressure_config["config"]["max_val"],
|
474 |
+
default_value=1022,
|
475 |
label="pressure")
|
476 |
|
477 |
minutes_of_day_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "minutes_of_day"), None)
|
|
|
479 |
minutes_of_day_slider = create_conditioning_slider(
|
480 |
min_val=minutes_of_day_config["config"]["min_val"],
|
481 |
max_val=minutes_of_day_config["config"]["max_val"],
|
482 |
+
default_value=1387,
|
483 |
label="minutes_of_day")
|
484 |
|
485 |
day_of_year_config = next((item for item in model_conditioning_config["configs"] if item["id"] == "day_of_year"), None)
|
|
|
487 |
day_of_year_slider = create_conditioning_slider(
|
488 |
min_val=day_of_year_config["config"]["min_val"],
|
489 |
max_val=day_of_year_config["config"]["max_val"],
|
490 |
+
default_value=254,
|
491 |
label="Day of year")
|
492 |
|
493 |
with gr.Accordion("Sampler params", open=False):
|