diff --git a/BaiHu/Tools/addModel.py b/BaiHu/Tools/addModel.py
index d28ccf95..04f7011f 100644
--- a/BaiHu/Tools/addModel.py
+++ b/BaiHu/Tools/addModel.py
@@ -3,22 +3,34 @@
 from Util.PgUtil import *
 
 # 修改操作（插入、更新或删除）
-sql = "INSERT INTO t_hy_model(model_id,model_name,memo,star,update_time,img_count,b_use,sort_id,model_type_id) VALUES (%s,%s, %s,%s,now(),%s,1,1,%s);"
+sql = "INSERT INTO t_hy_model(model_id,model_name,memo,star,update_time,img_count,b_use,sort_id,model_type_id,style_id) VALUES (%s,%s, %s,%s,now(),%s,1,1,%s,%s);"
 
 # 模型编号
 model_id = 28
+
 # 模型名称
 model_name = '火烈鸟'
+
 # 提示信息
 memo = '1、请上传正面半身照。; 2、确保脸部无遮挡；3、图片要求不小于1.5MB（正常手机拍照即可）'
+
 # 星级
 star = 5
+
 # 上传照片数量
 img_count = 1
+
 # 模板类型
 # 1:SD TXT2IMG  2:SD IMG2IMG 3:COMFY_UI
 model_type_id = 2
 
-params = (model_id, model_name, memo, star, img_count, model_type_id)
-execute_modify(sql, params)
+'''
+1	卡通风格
+2	儿童摄影
+3	风格写真
+4	双人系列
+'''
+style_id = 2
 
+params = (model_id, model_name, memo, star, img_count, model_type_id, style_id)
+execute_modify(sql, params)