mirror of https://github.com/coqui-ai/TTS.git
Fixup
This commit is contained in:
parent
623ea41634
commit
44b61d2b92
|
@ -307,8 +307,9 @@ class ModelManager(object):
|
||||||
if answer.lower() == "y":
|
if answer.lower() == "y":
|
||||||
with open(tos_path, "w") as f:
|
with open(tos_path, "w") as f:
|
||||||
f.write("I have read, understood ad agree the Terms and Conditions.")
|
f.write("I have read, understood ad agree the Terms and Conditions.")
|
||||||
|
return True
|
||||||
else:
|
else:
|
||||||
raise Exception("You must agree to the terms of service to use this model.")
|
return False
|
||||||
|
|
||||||
def tos_agreed(self, model_item, model_full_path):
|
def tos_agreed(self, model_item, model_full_path):
|
||||||
"""Check if the user has agreed to the terms of service"""
|
"""Check if the user has agreed to the terms of service"""
|
||||||
|
@ -342,7 +343,9 @@ class ModelManager(object):
|
||||||
os.makedirs(output_path, exist_ok=True)
|
os.makedirs(output_path, exist_ok=True)
|
||||||
# handle TOS
|
# handle TOS
|
||||||
if not self.tos_agreed(model_item, output_path):
|
if not self.tos_agreed(model_item, output_path):
|
||||||
self.ask_tos(output_path)
|
if not self.ask_tos(output_path):
|
||||||
|
os.rmdir(output_path)
|
||||||
|
raise Exception(" [!] You must agree to the terms of service to use this model.")
|
||||||
print(f" > Downloading model to {output_path}")
|
print(f" > Downloading model to {output_path}")
|
||||||
try:
|
try:
|
||||||
if "fairseq" in model_name:
|
if "fairseq" in model_name:
|
||||||
|
|
Loading…
Reference in New Issue