|
import os |
|
import warnings |
|
from config import models_path, results_path, sample_path |
|
from setup import setup_environment |
|
from synthesis import load_models |
|
from interface import create_interface |
|
|
|
warnings.filterwarnings("ignore") |
|
|
|
def main(): |
|
os.makedirs(models_path, exist_ok=True) |
|
os.makedirs(results_path, exist_ok=True) |
|
|
|
if (not os.path.exists(os.path.join(models_path, 'encoder.pt')) or |
|
not os.path.exists(os.path.join(models_path, 'synthesizer.pt')) or |
|
not os.path.exists(os.path.join(models_path, 'vocoder_HiFiGAN.pkl')) or |
|
not os.path.exists(sample_path)): |
|
setup_success = setup_environment() |
|
if not setup_success: |
|
print("Setup failed. Exiting.") |
|
exit(1) |
|
print("Setup completed successfully.") |
|
|
|
load_success = load_models() |
|
if not load_success: |
|
print("Failed to load models. Exiting.") |
|
exit(1) |
|
|
|
demo = create_interface() |
|
demo.launch() |
|
|
|
if __name__ == "__main__": |
|
main() |