Bark-Voice-Cloning / util /settings.py
kevinwang676's picture
Upload 4 files
580d62a
import yaml
class Settings:
def __init__(self, config_file):
self.config_file = config_file
self.load()
def load(self):
try:
with open(self.config_file, 'r') as f:
data = yaml.load(f, Loader=yaml.FullLoader)
self.selected_theme = data.get('selected_theme', "gstaff/xkcd")
self.server_name = data.get('server_name', "")
self.server_port = data.get('server_port', 0)
self.server_share = data.get('server_share', False)
self.input_text_desired_length = data.get('input_text_desired_length', 110)
self.input_text_max_length = data.get('input_text_max_length', 170)
self.silence_sentence = data.get('silence_between_sentences', 250)
self.silence_speakers = data.get('silence_between_speakers', 500)
self.output_folder_path = data.get('output_folder_path', 'outputs')
except:
self.selected_theme = "gstaff/xkcd"
def save(self):
data = {
'selected_theme': self.selected_theme,
'server_name': self.server_name,
'server_port': self.server_port,
'server_share': self.server_share,
'input_text_desired_length' : self.input_text_desired_length,
'input_text_max_length' : self.input_text_max_length,
'silence_between_sentences': self.silence_sentence,
'silence_between_speakers': self.silence_speakers,
'output_folder_path': self.output_folder_path
}
with open(self.config_file, 'w') as f:
yaml.dump(data, f)