fred-dev commited on
Commit
0a58b60
·
verified ·
1 Parent(s): 110dfa4

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 = (max_val + min_val) / 2
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):