Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -25,12 +25,12 @@ for model_id in models_id.split("\n"):
|
|
25 |
print(f"{model_id}!!!")
|
26 |
print("Models!!!")
|
27 |
print("PretrainedModels...")
|
28 |
-
|
29 |
-
snapshot_download(repo_id=model_id, local_dir=f"./PretrainedModels/{
|
30 |
print("PretrainedModels!!!")
|
31 |
list_files_tree("./")
|
32 |
-
cnhubert_base_path = f"./PretrainedModels/{
|
33 |
-
bert_path = f"./PretrainedModels/{
|
34 |
|
35 |
import gradio as gr
|
36 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
@@ -251,14 +251,17 @@ for folder_name in folder_names:
|
|
251 |
max_value = -1
|
252 |
max_file = None
|
253 |
gpt_path = f"{models_folder_path}/{folder_name}/GPT_weights"
|
254 |
-
|
255 |
-
|
256 |
-
|
257 |
-
|
258 |
-
|
259 |
-
|
260 |
-
|
261 |
-
|
|
|
|
|
|
|
262 |
data_path = f"{models_folder_path}/{folder_name}/Data"
|
263 |
models_info[speaker]["transcript"] = {}
|
264 |
wavs = []
|
|
|
25 |
print(f"{model_id}!!!")
|
26 |
print("Models!!!")
|
27 |
print("PretrainedModels...")
|
28 |
+
base_model_id = "None1145/GPT-SoVITS-Base"
|
29 |
+
snapshot_download(repo_id=model_id, local_dir=f"./PretrainedModels/{base_model_id}")
|
30 |
print("PretrainedModels!!!")
|
31 |
list_files_tree("./")
|
32 |
+
cnhubert_base_path = f"./PretrainedModels/{base_model_id}/chinese-hubert-base"
|
33 |
+
bert_path = f"./PretrainedModels/{base_model_id}/chinese-roberta-wwm-ext-large"
|
34 |
|
35 |
import gradio as gr
|
36 |
from transformers import AutoModelForMaskedLM, AutoTokenizer
|
|
|
251 |
max_value = -1
|
252 |
max_file = None
|
253 |
gpt_path = f"{models_folder_path}/{folder_name}/GPT_weights"
|
254 |
+
try:
|
255 |
+
for filename in os.listdir(gpt_path):
|
256 |
+
match = pattern.search(filename)
|
257 |
+
if match:
|
258 |
+
value = int(match.group(1))
|
259 |
+
if value > max_value:
|
260 |
+
max_value = value
|
261 |
+
max_file = filename
|
262 |
+
models_info[speaker]["gpt_weight"] = f"{gpt_path}/{max_file}"
|
263 |
+
except:
|
264 |
+
models_info[speaker]["gpt_weight"] = f"{base_model_id}/{GPT.ckpt}"
|
265 |
data_path = f"{models_folder_path}/{folder_name}/Data"
|
266 |
models_info[speaker]["transcript"] = {}
|
267 |
wavs = []
|