None1145 commited on
Commit
3ab0026
·
verified ·
1 Parent(s): 59d356b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -12
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
- model_id = "None1145/GPT-SoVITS-Base"
29
- snapshot_download(repo_id=model_id, local_dir=f"./PretrainedModels/{model_id}")
30
  print("PretrainedModels!!!")
31
  list_files_tree("./")
32
- cnhubert_base_path = f"./PretrainedModels/{model_id}/chinese-hubert-base"
33
- bert_path = f"./PretrainedModels/{model_id}/chinese-roberta-wwm-ext-large"
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
- for filename in os.listdir(gpt_path):
255
- match = pattern.search(filename)
256
- if match:
257
- value = int(match.group(1))
258
- if value > max_value:
259
- max_value = value
260
- max_file = filename
261
- models_info[speaker]["gpt_weight"] = f"{gpt_path}/{max_file}"
 
 
 
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 = []