File size: 1,977 Bytes
9829721 fbd43a1 ee326f0 9829721 e90b328 9829721 ee326f0 9829721 bebc2b8 9829721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
import logging
import platform
import gradio as gr
import log
from project_settings import environment, project_path, log_directory, time_zone_info
from tabs.vad_tab import get_vad_tab
from tabs.shell_tab import get_shell_tab
log.setup_size_rotating(log_directory=log_directory, tz_info=time_zone_info)
logger = logging.getLogger("main")
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--examples_dir",
# default=(project_path / "data").as_posix(),
default=(project_path / "data/examples").as_posix(),
type=str
)
parser.add_argument(
"--models_repo_id",
default="qgyd2021/cc_vad",
type=str
)
parser.add_argument(
"--trained_model_dir",
default=(project_path / "trained_models").as_posix(),
type=str
)
parser.add_argument(
"--hf_token",
default=environment.get("hf_token"),
type=str,
)
parser.add_argument(
"--server_port",
default=environment.get("server_port", 7860),
type=int
)
args = parser.parse_args()
return args
def main():
args = get_args()
# ui
with gr.Blocks() as blocks:
gr.Markdown(value="vad.")
with gr.Tabs():
_ = get_vad_tab(
trained_model_dir=args.trained_model_dir,
examples_dir=args.examples_dir,
models_repo_id=args.models_repo_id,
hf_token=args.hf_token,
)
_ = get_shell_tab()
# http://127.0.0.1:7866/
# http://10.75.27.247:7866/
blocks.queue().launch(
# share=True,
share=False if platform.system() == "Windows" else False,
server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
server_port=args.server_port,
show_error=True
)
return
if __name__ == "__main__":
main()
|