frontend/model_manage_page.py

@@ -7,7 +7,6 @@ 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是当前工作目录下的一个文件夹
@@ -16,7 +15,7 @@ def model_manage_page():
 
     with gr.Blocks() as demo:
         gr.Markdown("## 模型管理")
-        state_output = gr.Label(label="当前状态",value="当前未加载模型")  # 将 Textbox 改为 Label
+        output_text = gr.Textbox(label="当前状态", interactive=False)
         with gr.Row():
             with gr.Column(scale=3):
                 model_select_dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True)  # 将子文件夹列表添加到Dropdown组件中，并设置为可选
@@ -46,11 +45,11 @@ def model_manage_page():
                     max_seq_length=max_seq_length,
                     load_in_4bit=load_in_4bit,
                 )
-                return f"模型 {get_model_name(model)} 已加载"
+                return f"模型 {selected_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=state_output)
+        load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=output_text)
 
         def unload_model():
             try:
@@ -69,11 +68,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=state_output)
+        unload_button.click(fn=unload_model, inputs=None, outputs=output_text)
 
         def save_model(save_model_name):
             try:
@@ -89,7 +88,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=state_output)
+        save_button.click(fn=save_model, inputs=save_model_name_input, outputs=output_text)
 
         def refresh_model_list():
             try:

tools/__init__.py

@@ -1,4 +1,2 @@
 from .parse_markdown import parse_markdown
-from .scan_doc_dir import *
-from .json_example import generate_example_json
-from .model import get_model_name
+from .scan_doc_dir import *

tools/model.py

@@ -1,4 +0,0 @@
-import os
-
-def get_model_name(model):
-    return os.path.basename(model.name_or_path)