Update app.py

app.py

@@ -89,10 +89,7 @@ def append_log(logs: str, msg: str) -> str:
 # ==========================================
 
 def ensure_model(model_state: Dict, repo_id: str, device_map: str, dtype_name: str, logs: str):
-    """Ensure a model is available in model_state.
-    Only (re)load when repo_id changes or model_state is empty.
-    device_map/dtype_name 变更不会触发重新加载（按你的要求）。
-    """
+
     ms = model_state or {"repo_id": None, "tok": None, "model": None}
     if ms.get("repo_id") != repo_id or ms.get("model") is None:
         logs = append_log(logs, f"加载模型 {repo_id}（触发：repo 变更）…")
@@ -114,10 +111,7 @@ def predict(user_text: str,
             max_new_token: int, top_k: int,
             logs_state: str,
             model_state: Dict):
-    """Generator for streaming output + live logs.
-    Only reload when repo_id changes.
-    Expects custom model.chat(conversations, tokenizer, max_new_token=..., top_k=...).
-    """
+    
     messages_state = messages_state or []
     logs_state = logs_state or ""
 
@@ -181,11 +175,6 @@ def preload_on_repo_change(repo_id: str, device_map: str, dtype_name: str, logs_
 # Gradio UI
 # ==========================================
 with gr.Blocks(title="mini-moe Chat (Gradio)") as demo:
-    gr.Markdown("""
-    # 🤖 mini-moe Chat UI (Gradio)
-    仅在 **repo 变更** 时重新加载模型；设备/精度变更不会触发重新加载（按你的要求）。
-    右侧含 **日志面板**，实时显示加载与推理步骤；**不使用 system prompt**。
-    """)
 
     messages_state = gr.State([])  # 保存 role/content 历史
     logs_state = gr.State("")     # 保存日志文本