main.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
import platform

import gradio as gr

from project_settings import environment, project_path


def get_args():
    parser = argparse.ArgumentParser()
    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="in progress.")

    # http://127.0.0.1:7864/
    blocks.queue().launch(
        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
    )
    return


if __name__ == "__main__":
    main()