forked from mrq/ai-voice-cloning
40 lines
1.2 KiB
Python
Executable File
40 lines
1.2 KiB
Python
Executable File
import os
|
|
|
|
if 'TORTOISE_MODELS_DIR' not in os.environ:
|
|
os.environ['TORTOISE_MODELS_DIR'] = os.path.realpath(os.path.join(os.getcwd(), './models/tortoise/'))
|
|
|
|
if 'TRANSFORMERS_CACHE' not in os.environ:
|
|
os.environ['TRANSFORMERS_CACHE'] = os.path.realpath(os.path.join(os.getcwd(), './models/transformers/'))
|
|
|
|
os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'python'
|
|
|
|
from utils import *
|
|
from webui import *
|
|
|
|
if __name__ == "__main__":
|
|
args = setup_args()
|
|
|
|
if args.listen_path is not None and args.listen_path != "/":
|
|
import uvicorn
|
|
uvicorn.run("main:app", host=args.listen_host, port=args.listen_port if not None else 8000)
|
|
else:
|
|
webui = setup_gradio()
|
|
webui.launch(share=args.share, prevent_thread_lock=True, show_error=True, server_name=args.listen_host, server_port=args.listen_port)
|
|
if not args.defer_tts_load:
|
|
tts = load_tts()
|
|
|
|
webui.block_thread()
|
|
elif __name__ == "main":
|
|
from fastapi import FastAPI
|
|
import gradio as gr
|
|
|
|
import sys
|
|
sys.argv = [sys.argv[0]]
|
|
|
|
app = FastAPI()
|
|
args = setup_args()
|
|
webui = setup_gradio()
|
|
app = gr.mount_gradio_app(app, webui, path=args.listen_path)
|
|
|
|
if not args.defer_tts_load:
|
|
tts = load_tts() |