Rename app (3).py to app.py
Browse files- app (3).py → app.py +20 -20
app (3).py → app.py
RENAMED
|
@@ -3,12 +3,12 @@ import os
|
|
| 3 |
import shutil
|
| 4 |
import subprocess
|
| 5 |
|
| 6 |
-
# ----
|
| 7 |
def upload_audio(audio_files, model_name):
|
| 8 |
if not audio_files:
|
| 9 |
-
return "⚠️
|
| 10 |
if not model_name:
|
| 11 |
-
return "⚠️ Model
|
| 12 |
|
| 13 |
save_dir = os.path.join("dataset", model_name)
|
| 14 |
os.makedirs(save_dir, exist_ok=True)
|
|
@@ -16,13 +16,13 @@ def upload_audio(audio_files, model_name):
|
|
| 16 |
for i, file in enumerate(audio_files):
|
| 17 |
shutil.copy(file.name, os.path.join(save_dir, f"{model_name}_{i}.wav"))
|
| 18 |
|
| 19 |
-
return f"✅ {len(audio_files)}
|
| 20 |
|
| 21 |
-
# ----
|
| 22 |
def train_rvc(model_name, sample_rate, epochs):
|
| 23 |
dataset_dir = os.path.join("dataset", model_name)
|
| 24 |
if not os.path.exists(dataset_dir):
|
| 25 |
-
return f"⚠️ {dataset_dir}
|
| 26 |
|
| 27 |
try:
|
| 28 |
cmd = [
|
|
@@ -33,28 +33,28 @@ def train_rvc(model_name, sample_rate, epochs):
|
|
| 33 |
"--epochs", str(epochs)
|
| 34 |
]
|
| 35 |
subprocess.Popen(cmd)
|
| 36 |
-
return f"🚀
|
| 37 |
except Exception as e:
|
| 38 |
-
return f"❌
|
| 39 |
|
| 40 |
-
# ----
|
| 41 |
with gr.Blocks() as demo:
|
| 42 |
-
gr.Markdown("# 🎙️ RVC
|
| 43 |
-
gr.Markdown("1️⃣
|
| 44 |
|
| 45 |
-
with gr.Tab("
|
| 46 |
with gr.Row():
|
| 47 |
-
audio_files = gr.File(file_count="multiple", label="🎧
|
| 48 |
-
model_name = gr.Textbox(label="Model
|
| 49 |
-
output_upload = gr.Textbox(label="
|
| 50 |
-
upload_button = gr.Button("📦
|
| 51 |
upload_button.click(upload_audio, inputs=[audio_files, model_name], outputs=output_upload)
|
| 52 |
|
| 53 |
-
with gr.Tab("
|
| 54 |
sample_rate = gr.Dropdown(choices=[32000, 40000, 48000], value=40000, label="Sample Rate (Hz)")
|
| 55 |
-
epochs = gr.Slider(50, 1000, value=750, step=50, label="
|
| 56 |
-
output_train = gr.Textbox(label="
|
| 57 |
-
train_button = gr.Button("🚀
|
| 58 |
train_button.click(train_rvc, inputs=[model_name, sample_rate, epochs], outputs=output_train)
|
| 59 |
|
| 60 |
demo.launch()
|
|
|
|
| 3 |
import shutil
|
| 4 |
import subprocess
|
| 5 |
|
| 6 |
+
# ---- Upload audio function ----
|
| 7 |
def upload_audio(audio_files, model_name):
|
| 8 |
if not audio_files:
|
| 9 |
+
return "⚠️ Please upload at least one audio file."
|
| 10 |
if not model_name:
|
| 11 |
+
return "⚠️ Model name cannot be empty."
|
| 12 |
|
| 13 |
save_dir = os.path.join("dataset", model_name)
|
| 14 |
os.makedirs(save_dir, exist_ok=True)
|
|
|
|
| 16 |
for i, file in enumerate(audio_files):
|
| 17 |
shutil.copy(file.name, os.path.join(save_dir, f"{model_name}_{i}.wav"))
|
| 18 |
|
| 19 |
+
return f"✅ {len(audio_files)} audio files saved to '{save_dir}' folder!"
|
| 20 |
|
| 21 |
+
# ---- Start training function ----
|
| 22 |
def train_rvc(model_name, sample_rate, epochs):
|
| 23 |
dataset_dir = os.path.join("dataset", model_name)
|
| 24 |
if not os.path.exists(dataset_dir):
|
| 25 |
+
return f"⚠️ {dataset_dir} folder not found. Upload audio files first."
|
| 26 |
|
| 27 |
try:
|
| 28 |
cmd = [
|
|
|
|
| 33 |
"--epochs", str(epochs)
|
| 34 |
]
|
| 35 |
subprocess.Popen(cmd)
|
| 36 |
+
return f"🚀 Training started! Model name: {model_name}, Epochs: {epochs}, Sample rate: {sample_rate}"
|
| 37 |
except Exception as e:
|
| 38 |
+
return f"❌ Training failed to start: {e}"
|
| 39 |
|
| 40 |
+
# ---- Interface ----
|
| 41 |
with gr.Blocks() as demo:
|
| 42 |
+
gr.Markdown("# 🎙️ RVC Voice Model Training Tool")
|
| 43 |
+
gr.Markdown("1️⃣ Upload audio → 2️⃣ Enter model name → 3️⃣ Start training")
|
| 44 |
|
| 45 |
+
with gr.Tab("Upload Audio"):
|
| 46 |
with gr.Row():
|
| 47 |
+
audio_files = gr.File(file_count="multiple", label="🎧 Upload Audio Files (.wav)")
|
| 48 |
+
model_name = gr.Textbox(label="Model Name", placeholder="e.g.: zeynep_rvc")
|
| 49 |
+
output_upload = gr.Textbox(label="Status")
|
| 50 |
+
upload_button = gr.Button("📦 Upload Audio")
|
| 51 |
upload_button.click(upload_audio, inputs=[audio_files, model_name], outputs=output_upload)
|
| 52 |
|
| 53 |
+
with gr.Tab("Start Training"):
|
| 54 |
sample_rate = gr.Dropdown(choices=[32000, 40000, 48000], value=40000, label="Sample Rate (Hz)")
|
| 55 |
+
epochs = gr.Slider(50, 1000, value=750, step=50, label="Number of Epochs")
|
| 56 |
+
output_train = gr.Textbox(label="Status")
|
| 57 |
+
train_button = gr.Button("🚀 Start Training")
|
| 58 |
train_button.click(train_rvc, inputs=[model_name, sample_rate, epochs], outputs=output_train)
|
| 59 |
|
| 60 |
demo.launch()
|