From b6bc17f7305dc80f03e6021b59fa49c844f4f3ca Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Thu, 28 Nov 2024 22:23:41 -0500
Subject: [PATCH] [U] Compute hostname differently

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

diff --git a/src/llamafactory/webui/interface.py b/src/llamafactory/webui/interface.py
index 51e1d7d3..a4f2c1fa 100644
--- a/src/llamafactory/webui/interface.py
+++ b/src/llamafactory/webui/interface.py
@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import os
-import socket
+import platform
 
 from ..extras.packages import is_gradio_available
 from .common import save_config
@@ -35,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=f"LLaMA Board ({socket.gethostname()})", css=CSS) as demo:
+    with gr.Blocks(title=f"LLaMA Board ({hostname})", 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(