From 0fa2b51a799ed0c4fa58cf64d737ea4b85542dbf Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Fri, 11 Apr 2025 00:14:40 +0800 Subject: [PATCH] =?UTF-8?q?refactor(frontend):=20=E4=BC=98=E5=8C=96?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E7=AE=A1=E7=90=86=E9=A1=B5=E9=9D=A2=E7=9A=84?= =?UTF-8?q?=E4=BA=A4=E4=BA=92=E5=92=8C=E6=98=BE=E7=A4=BA?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 将状态输出从 Textbox 改为 Label 组件，提高用户体验 - 添加 get_model_name 函数以获取模型名称，提高代码复用性 - 更新模型加载、卸载和保存后的状态显示，使信息更加准确 - 优化模型列表刷新功能，确保模型列表实时更新 --- frontend/model_manage_page.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/frontend/model_manage_page.py b/frontend/model_manage_page.py index 5466da2..aa3b302 100644 --- a/frontend/model_manage_page.py +++ b/frontend/model_manage_page.py @@ -7,6 +7,7 @@ import torch sys.path.append(str(Path(__file__).resolve().parent.parent)) from global_var import model,tokenizer +from tools.model import get_model_name def model_manage_page(): workdir = "workdir" # 假设workdir是当前工作目录下的一个文件夹 @@ -15,7 +16,7 @@ def model_manage_page(): with gr.Blocks() as demo: gr.Markdown("## 模型管理") - output_text = gr.Textbox(label="当前状态", interactive=False) + state_output = gr.Label(label="当前状态",value="当前未加载模型") # 将 Textbox 改为 Label with gr.Row(): with gr.Column(scale=3): model_select_dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True) # 将子文件夹列表添加到Dropdown组件中，并设置为可选 @@ -45,11 +46,11 @@ def model_manage_page(): max_seq_length=max_seq_length, load_in_4bit=load_in_4bit, ) - return f"模型 {selected_model} 已加载" + return f"模型 {get_model_name(model)} 已加载" except Exception as e: return f"加载模型时出错: {str(e)}" - load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=output_text) + load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=state_output) def unload_model(): try: @@ -68,11 +69,11 @@ def model_manage_page(): # 将模型设置为 None model = None - return "模型已卸载" + return "当前未加载模型" except Exception as e: return f"卸载模型时出错: {str(e)}" - unload_button.click(fn=unload_model, inputs=None, outputs=output_text) + unload_button.click(fn=unload_model, inputs=None, outputs=state_output) def save_model(save_model_name): try: @@ -88,7 +89,7 @@ def model_manage_page(): except Exception as e: return f"保存模型时出错: {str(e)}" - save_button.click(fn=save_model, inputs=save_model_name_input, outputs=output_text) + save_button.click(fn=save_model, inputs=save_model_name_input, outputs=state_output) def refresh_model_list(): try: