diff --git a/src/llamafactory/webui/interface.py b/src/llamafactory/webui/interface.py index 55aed5e5..a4f2c1fa 100644 --- a/src/llamafactory/webui/interface.py +++ b/src/llamafactory/webui/interface.py @@ -13,6 +13,7 @@ # limitations under the License. import os +import platform from ..extras.packages import is_gradio_available from .common import save_config @@ -34,8 +35,9 @@ if is_gradio_available(): def create_ui(demo_mode: bool = False) -> "gr.Blocks": engine = Engine(demo_mode=demo_mode, pure_chat=False) + hostname = os.getenv("HOSTNAME", os.getenv("COMPUTERNAME", platform.node())).split('.')[0] - with gr.Blocks(title="LLaMA Board", css=CSS) as demo: + with gr.Blocks(title=f"LLaMA Board ({hostname})", css=CSS) as demo: if demo_mode: gr.HTML("