diff --git a/frontend/model_manage_page.py b/frontend/model_manage_page.py
index fe2f0ce..7483049 100644
--- a/frontend/model_manage_page.py
+++ b/frontend/model_manage_page.py
@@ -7,7 +7,7 @@ import torch
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 from global_var import get_model, get_tokenizer, set_model, set_tokenizer
-from tools.model import get_model_name
+from train import get_model_name
 
 def model_manage_page():
     workdir = "workdir"  # 假设workdir是当前工作目录下的一个文件夹
diff --git a/frontend/train_page.py b/frontend/train_page.py
index bcfd986..91e0e27 100644
--- a/frontend/train_page.py
+++ b/frontend/train_page.py
@@ -8,7 +8,8 @@ from transformers import TrainerCallback
 
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 from global_var import get_model, get_tokenizer, get_datasets, get_workdir
-from tools import train_model, find_available_port
+from tools import find_available_port
+from train import train_model
 
 def train_page():
     with gr.Blocks() as demo:
diff --git a/main.py b/main.py
index f44429a..e7337f1 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,5 @@
 import gradio as gr
-import unsloth
+import train
 from frontend import *
 from db import initialize_sqlite_db, initialize_prompt_store
 from global_var import init_global_var, get_sql_engine, get_prompt_store
diff --git a/tools/__init__.py b/tools/__init__.py
index 6d15436..b384ee1 100644
--- a/tools/__init__.py
+++ b/tools/__init__.py
@@ -1,6 +1,5 @@
-from .parse_markdown import parse_markdown
+from .parse_markdown import *
 from .document import *
 from .json_example import generate_example_json
-from .model import *
 from .port import *
 from .reasoning import call_openai_api
\ No newline at end of file
diff --git a/train/__init__.py b/train/__init__.py
new file mode 100644
index 0000000..71a1d85
--- /dev/null
+++ b/train/__init__.py
@@ -0,0 +1 @@
+from .model import *
\ No newline at end of file
diff --git a/tools/model.py b/train/model.py
similarity index 100%
rename from tools/model.py
rename to train/model.py