Compare commits

...

2 Commits

Author SHA1 Message Date
carry
2e552c186d refactor(frontend): 重构模型选择界面的变量命名
- 将模型选择的 Dropdown 组件从 dropdown 重命名为 model_select_dropdown,提高代码可读性
- 更新 load_button 和 refresh_button 的输出目标,以适应新的变量名
2025-04-10 21:19:58 +08:00
carry
1b3f546669 refactor(frontend): 重构前端页面并添加独立运行功能
- 在 chat_page 和 prompt_manage_page 中添加了独立运行的入口
- 引入 sys 和 pathlib 模块以支持路径操作
- 修改了模块导入方式,使其能够作为独立脚本运行
- 优化了代码结构,提高了可读性和可维护性
2025-04-10 21:18:05 +08:00
3 changed files with 21 additions and 5 deletions

View File

@ -1,4 +1,7 @@
import gradio as gr
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import model,tokenizer
def chat_page():
@ -23,6 +26,10 @@ def chat_page():
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
bot, chatbot, chatbot
)
clear.click(lambda: None, None, chatbot, queue=False)
return demo
return demo
if __name__ == "__main__":
chat_page().queue().launch()

View File

@ -18,7 +18,7 @@ def model_manage_page():
output_text = gr.Textbox(label="当前状态", interactive=False)
with gr.Row():
with gr.Column(scale=3):
dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True) # 将子文件夹列表添加到Dropdown组件中并设置为可选
model_select_dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True) # 将子文件夹列表添加到Dropdown组件中并设置为可选
max_seq_length_input = gr.Number(label="最大序列长度", value=4096, precision=0)
load_in_4bit_input = gr.Checkbox(label="使用4位量化", value=True)
with gr.Column(scale=1):
@ -49,7 +49,7 @@ def model_manage_page():
except Exception as e:
return f"加载模型时出错: {str(e)}"
load_button.click(fn=load_model, inputs=[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=output_text)
def unload_model():
try:
@ -98,7 +98,7 @@ def model_manage_page():
except Exception as e:
return f"刷新模型列表时出错: {str(e)}"
refresh_button.click(fn=refresh_model_list, inputs=None, outputs=dropdown)
refresh_button.click(fn=refresh_model_list, inputs=None, outputs=model_select_dropdown)
return demo

View File

@ -1,5 +1,8 @@
import gradio as gr
import sys
from pathlib import Path
from typing import List
sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import prompt_store
from schema.prompt import promptTempleta
def prompt_manage_page():
@ -113,4 +116,10 @@ def prompt_manage_page():
outputs=[prompt_table]
)
return demo
return demo
if __name__ == "__main__":
demo = prompt_manage_page()
demo.queue()
demo.launch()