Update app.py
Browse files
app.py
CHANGED
|
@@ -15,12 +15,12 @@ with gr.Blocks() as demo:
|
|
| 15 |
with gr.Row():
|
| 16 |
with gr.Accordion("Inference Settings"):
|
| 17 |
pitch = gr.Slider(minimum=-12, maximum=12, step=1, label="Pitch", value=0)
|
| 18 |
-
f0_method = gr.Dropdown(choices=["rmvpe", "pm", "harvest"], label="f0 Method", value="rmvpe")
|
| 19 |
-
index_rate = gr.Slider(minimum=0, maximum=1, step=0.01, label="Index Rate", value=0.5)
|
| 20 |
-
volume_normalization = gr.Slider(minimum=0, maximum=1, step=0.01, label="Volume Normalization", value=0)
|
| 21 |
-
consonant_protection = gr.Slider(minimum=0, maximum=1, step=0.01, label="Consonant Protection", value=0.5)
|
| 22 |
with gr.Row():
|
| 23 |
-
save_as = gr.Textbox(value="/content/RVC/audios/output_audio.wav", label="Output Audio Path")
|
| 24 |
|
| 25 |
run_btn = gr.Button("Run Inference")
|
| 26 |
with gr.Row():
|
|
@@ -32,7 +32,7 @@ with gr.Blocks() as demo:
|
|
| 32 |
with gr.TabItem("Create Index and stuff"):
|
| 33 |
model_name = gr.Textbox(label="Model Name (No spaces or symbols)")
|
| 34 |
dataset_folder = gr.Textbox(label="Dataset Folder", value="/content/dataset")
|
| 35 |
-
f0method = gr.Dropdown(["pm", "harvest", "rmvpe", "rmvpe_gpu"], label="F0 Method", value="rmvpe_gpu")
|
| 36 |
preprocess_btn = gr.Button("Start Preprocessing")
|
| 37 |
f0_btn = gr.Button("Extract F0 Feature")
|
| 38 |
train_btn = gr.Button("Train Index")
|
|
@@ -45,9 +45,9 @@ with gr.Blocks() as demo:
|
|
| 45 |
#train_btn.click(train_index, inputs=[model_name, "v2"], outputs=train_output)
|
| 46 |
with gr.TabItem("Train Your Model"):
|
| 47 |
model_name_input = gr.Textbox(label="Model Name", placeholder="Enter the model name", interactive=True)
|
| 48 |
-
epochs_slider = gr.Slider(minimum=50, maximum=2000, value=200, step=10, label="Epochs")
|
| 49 |
-
save_frequency_slider = gr.Slider(minimum=10, maximum=100, value=50, step=10, label="Save Frequency")
|
| 50 |
-
batch_size_slider = gr.Slider(minimum=1, maximum=20, value=8, step=1, label="Batch Size")
|
| 51 |
|
| 52 |
train_button = gr.Button("Train Model")
|
| 53 |
training_output = gr.Textbox(label="Training Log", interactive=False)
|
|
|
|
| 15 |
with gr.Row():
|
| 16 |
with gr.Accordion("Inference Settings"):
|
| 17 |
pitch = gr.Slider(minimum=-12, maximum=12, step=1, label="Pitch", value=0)
|
| 18 |
+
f0_method = gr.Dropdown(choices=["rmvpe", "pm", "harvest"], label="f0 Method", value="rmvpe", interactive=True)
|
| 19 |
+
index_rate = gr.Slider(minimum=0, maximum=1, step=0.01, label="Index Rate", value=0.5, interactive=True)
|
| 20 |
+
volume_normalization = gr.Slider(minimum=0, maximum=1, step=0.01, label="Volume Normalization", value=0, interactive=True)
|
| 21 |
+
consonant_protection = gr.Slider(minimum=0, maximum=1, step=0.01, label="Consonant Protection", value=0.5, interactive=True)
|
| 22 |
with gr.Row():
|
| 23 |
+
save_as = gr.Textbox(value="/content/RVC/audios/output_audio.wav", label="Output Audio Path", interactive=True)
|
| 24 |
|
| 25 |
run_btn = gr.Button("Run Inference")
|
| 26 |
with gr.Row():
|
|
|
|
| 32 |
with gr.TabItem("Create Index and stuff"):
|
| 33 |
model_name = gr.Textbox(label="Model Name (No spaces or symbols)")
|
| 34 |
dataset_folder = gr.Textbox(label="Dataset Folder", value="/content/dataset")
|
| 35 |
+
f0method = gr.Dropdown(["pm", "harvest", "rmvpe", "rmvpe_gpu"], label="F0 Method", value="rmvpe_gpu", interactive=True)
|
| 36 |
preprocess_btn = gr.Button("Start Preprocessing")
|
| 37 |
f0_btn = gr.Button("Extract F0 Feature")
|
| 38 |
train_btn = gr.Button("Train Index")
|
|
|
|
| 45 |
#train_btn.click(train_index, inputs=[model_name, "v2"], outputs=train_output)
|
| 46 |
with gr.TabItem("Train Your Model"):
|
| 47 |
model_name_input = gr.Textbox(label="Model Name", placeholder="Enter the model name", interactive=True)
|
| 48 |
+
epochs_slider = gr.Slider(minimum=50, maximum=2000, value=200, step=10, label="Epochs", interactive=True)
|
| 49 |
+
save_frequency_slider = gr.Slider(minimum=10, maximum=100, value=50, step=10, label="Save Frequency", interactive=True)
|
| 50 |
+
batch_size_slider = gr.Slider(minimum=1, maximum=20, value=8, step=1, label="Batch Size", interactive=True)
|
| 51 |
|
| 52 |
train_button = gr.Button("Train Model")
|
| 53 |
training_output = gr.Textbox(label="Training Log", interactive=False)
|