From f5e6e25a1b25dd03dea6e2fe6a4b3e6425cc1f6b Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Thu, 28 Nov 2024 12:25:02 +0800
Subject: [PATCH] [+] Show the hostname

Former-commit-id: dfb953b1adb79ed9f886ecac6851a559617aaa5a
---
 src/llamafactory/webui/interface.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/llamafactory/webui/interface.py b/src/llamafactory/webui/interface.py
index 55aed5e5..51e1d7d3 100644
--- a/src/llamafactory/webui/interface.py
+++ b/src/llamafactory/webui/interface.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import os
+import socket
 
 from ..extras.packages import is_gradio_available
 from .common import save_config
@@ -35,7 +36,7 @@ if is_gradio_available():
 def create_ui(demo_mode: bool = False) -> "gr.Blocks":
     engine = Engine(demo_mode=demo_mode, pure_chat=False)
 
-    with gr.Blocks(title="LLaMA Board", css=CSS) as demo:
+    with gr.Blocks(title=f"LLaMA Board ({socket.gethostname()})", css=CSS) as demo:
         if demo_mode:
             gr.HTML("<h1><center>LLaMA Board: A One-stop Web UI for Getting Started with LLaMA Factory</center></h1>")
             gr.HTML(