diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py index e048d780..3969b799 100644 --- a/tabs/inference/inference.py +++ b/tabs/inference/inference.py @@ -39,7 +39,10 @@ os.path.join(root, file) for root, _, files in os.walk(model_root, topdown=False) for file in files - if file.endswith((".pth", ".onnx")) + if ( + file.endswith((".pth", ".onnx")) + and not (file.startswith("G_") or file.startswith("D_")) + ) ] indexes_list = [ @@ -73,7 +76,10 @@ def change_choices(): os.path.join(root, file) for root, _, files in os.walk(model_root, topdown=False) for file in files - if file.endswith((".pth", ".onnx")) + if ( + file.endswith((".pth", ".onnx")) + and not (file.startswith("G_") or file.startswith("D_")) + ) ] indexes_list = [ @@ -262,7 +268,14 @@ def inference_tab(): value=False, interactive=True, ) - pitch = gr.Slider(-12, 12, 0, label=i18n("Pitch")) + pitch = gr.Slider( + minimum=-12, + maximum=12, + step=1, + label=i18n("Pitch"), + value=0, + interactive=True, + ) filter_radius = gr.Slider( minimum=0, maximum=7, diff --git a/tabs/tts/tts.py b/tabs/tts/tts.py index 7c70fecd..749e488e 100644 --- a/tabs/tts/tts.py +++ b/tabs/tts/tts.py @@ -39,7 +39,10 @@ os.path.join(root, file) for root, _, files in os.walk(model_root, topdown=False) for file in files - if file.endswith((".pth", ".onnx")) + if ( + file.endswith((".pth", ".onnx")) + and not (file.startswith("G_") or file.startswith("D_")) + ) ] indexes_list = [ @@ -64,7 +67,10 @@ def change_choices(): os.path.join(root, file) for root, _, files in os.walk(model_root, topdown=False) for file in files - if file.endswith((".pth", ".onnx")) + if ( + file.endswith((".pth", ".onnx")) + and not (file.startswith("G_") or file.startswith("D_")) + ) ] indexes_list = [ @@ -251,7 +257,14 @@ def tts_tab(): interactive=True, ) - pitch = gr.Slider(-12, 12, 0, label=i18n("Pitch")) + pitch = gr.Slider( + minimum=-12, + maximum=12, + step=1, + label=i18n("Pitch"), + value=0, + interactive=True, + ) filter_radius = gr.Slider( minimum=0, maximum=7,