Spaces:
Configuration error
Configuration error
File size: 5,108 Bytes
8bb5911 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
import gradio as gr
from functools import lru_cache
import logging, os
from tabs.typing_extra import generate_switch, RVC_MODELS_DIR, update_models_list, show_hop_slider
@lru_cache(maxsize=1)
def get_current_models(models_dir):
"""Retrieve list of model directories, excluding specific items."""
try:
models_list = [item for item in os.listdir(models_dir) if item != 'mute' and os.path.isdir(os.path.join(models_dir, item))]
return sorted(models_list)
except OSError as e:
logger.error(f"Error accessing models directory: {e}\n{format_exc()}")
raise gr.Error(f"Failed to list models. Check directory permissions: {e}")
def inference_tab(status_message, rvc_model):
"""Define the Inference tab UI."""
with gr.TabItem("Inference"):
with gr.Row(equal_height=True):
rvc_model = gr.Dropdown(
label="Voice Model",
info="Select a model for voice conversion.",
choices=get_current_models(RVC_MODELS_DIR),
allow_custom_value=False
)
pitch = gr.Slider(
minimum=-12,
maximum=12,
value=0,
step=1,
label="Pitch Change (Vocals Only)",
info="Adjust vocal pitch in semitones."
)
with gr.Row(equal_height=True):
refresh_btn = gr.Button("Refresh Models", variant="secondary")
refresh_btn.click(
fn=update_models_list,
outputs=[rvc_model, status_message]
)
with gr.Group():
with gr.Row(equal_height=True):
with gr.Column(visible=True, variant="panel", scale=1):
song_input = gr.Textbox(
label="Song Input",
placeholder="Enter a YouTube link or local file path",
lines=1
)
with gr.Accordion("RVC Settings", open=False):
with gr.Row(equal_height=True):
with gr.Column(variant="panel", scale=1):
f0_method = gr.Dropdown(
choices=[
"rmvpe", "crepe", "fcpe",
"hybrid[rmvpe+fcpe]", "crepe-tiny"
],
value="rmvpe",
label="Pitch Detection Algorithm",
info="Algorithm for detecting vocal pitch."
)
with gr.Row(equal_height=True):
filter_radius = gr.Slider(0, 7, value=3, step=1, label="Filter Radius")
rms_mix_rate = gr.Slider(0, 1, value=0.25, step=0.01, label="RMS Mix Rate")
protect = gr.Slider(0, 0.5, value=0.33, step=0.01, label="Protect Rate")
index_rate = gr.Slider(0, 1, value=0.5, step=0.01, label="Search Feature Ratio")
crepe_hop_length = gr.Slider(
32, 320, value=128, step=1,
label="Crepe Hop Length",
visible=False,
info="Hop length for crepe pitch detection."
)
f0_method.change(
fn=show_hop_slider,
inputs=f0_method,
outputs=crepe_hop_length
)
with gr.Accordion("Audio Settings", open=False):
with gr.Column(variant="panel", scale=1):
with gr.Row(equal_height=True):
main_gain = gr.Slider(-20, 20, value=0, step=1, label="Main Vocals Gain (dB)")
backup_gain = gr.Slider(-20, 20, value=0, step=1, label="Backup Vocals Gain (dB)")
inst_gain = gr.Slider(-20, 20, value=0, step=1, label="Instrumental Gain (dB)")
with gr.Row(equal_height=True):
keep_files = gr.Checkbox(label="Keep Intermediate Files", value=True)
vocal_only = gr.Checkbox(label="Vocal-Only Conversion", value=False)
with gr.Row(equal_height=True):
generate_btn = gr.Button("Convert", variant="primary", scale=2)
ai_cover = gr.Audio(label="Output Cover", type="filepath", interactive=False, scale=9)
with gr.Column(min_width=160):
output_format = gr.Dropdown(
choices=["mp3", "wav", "flac"],
value="mp3",
label="Output Format"
)
generate_btn.click(
fn=generate_switch,
inputs=[
song_input, rvc_model, pitch, keep_files, main_gain,
backup_gain, inst_gain, index_rate, filter_radius, rms_mix_rate,
f0_method, crepe_hop_length, protect, output_format, vocal_only
],
outputs=[ai_cover, status_message],
queue=True,
show_progress="full"
)
return rvc_model
|