Update demo.py
Browse files
demo.py
CHANGED
|
@@ -175,15 +175,14 @@ with gr.Blocks() as demo:
|
|
| 175 |
output_audio = gr.Audio(label="Output Audio",interactive=False)
|
| 176 |
run_btn.click(run_inference, [model_name, pitch, input_path, f0_method, save_as, index_rate, volume_normalization, consonant_protection], output_message)
|
| 177 |
|
| 178 |
-
with gr.Tab("Training")
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
preprocess_output = gr.Textbox(label="Preprocessing Log")
|
| 188 |
f0_output = gr.Textbox(label="F0 Feature Extraction Log")
|
| 189 |
train_output = gr.Textbox(label="Training Log")
|
|
|
|
| 175 |
output_audio = gr.Audio(label="Output Audio",interactive=False)
|
| 176 |
run_btn.click(run_inference, [model_name, pitch, input_path, f0_method, save_as, index_rate, volume_normalization, consonant_protection], output_message)
|
| 177 |
|
| 178 |
+
with gr.Tab("Training"):\
|
| 179 |
+
model_name = gr.Textbox(label="Model Name (No spaces or symbols)")
|
| 180 |
+
dataset_folder = gr.Textbox(label="Dataset Folder", value="/content/dataset")
|
| 181 |
+
f0method = gr.Dropdown(["pm", "harvest", "rmvpe", "rmvpe_gpu"], label="F0 Method", value="rmvpe_gpu")
|
| 182 |
|
| 183 |
+
preprocess_btn = gr.Button("Start Preprocessing")
|
| 184 |
+
f0_btn = gr.Button("Extract F0 Feature")
|
| 185 |
+
train_btn = gr.Button("Train Index")
|
|
|
|
| 186 |
preprocess_output = gr.Textbox(label="Preprocessing Log")
|
| 187 |
f0_output = gr.Textbox(label="F0 Feature Extraction Log")
|
| 188 |
train_output = gr.Textbox(label="Training Log")
|