Zeph27 commited on
Commit
da92eeb
·
1 Parent(s): 7393793
Files changed (1) hide show
  1. src/webui.py +5 -4
src/webui.py CHANGED
@@ -8,6 +8,7 @@ from argparse import ArgumentParser
8
 
9
  import gradio as gr
10
  import spaces
 
11
 
12
  from main import song_cover_pipeline
13
 
@@ -21,11 +22,11 @@ def download_and_extract_model(model_url, model_name, progress=gr.Progress()):
21
  try:
22
  os.makedirs(rvc_models_dir, exist_ok=True)
23
 
24
- extraction_folder = os.path.join(rvc_models_dir, model_name)
25
- zip_path = os.path.join(rvc_models_dir, f'{model_name}.zip')
26
-
27
  if os.path.exists(extraction_folder):
28
- raise gr.Error(f'Voice model directory {model_name} already exists! Choose a different name for your voice model.')
 
 
 
29
 
30
  progress(0, desc=f'[~] Downloading voice model with name {model_name}...')
31
 
 
8
 
9
  import gradio as gr
10
  import spaces
11
+ import random
12
 
13
  from main import song_cover_pipeline
14
 
 
22
  try:
23
  os.makedirs(rvc_models_dir, exist_ok=True)
24
 
 
 
 
25
  if os.path.exists(extraction_folder):
26
+ model_name = model_name + str(random.randint(0, 1000))
27
+
28
+ extraction_folder = os.path.join(rvc_models_dir, model_name)
29
+ zip_path = os.path.join(rvc_models_dir, f'{model_name}.zip')
30
 
31
  progress(0, desc=f'[~] Downloading voice model with name {model_name}...')
32