diff --git a/flask_app/routes/judge_zbfile.py b/flask_app/routes/judge_zbfile.py index b34df9b..ba43a31 100644 --- a/flask_app/routes/judge_zbfile.py +++ b/flask_app/routes/judge_zbfile.py @@ -36,7 +36,7 @@ def judge_zbfile() -> Any: #判断是否是招标文件 # ) result = pool.apply( judge_zbfile_exec, # 你的实际执行函数 - args=(file_path,logger,) + args=(file_path,) ) return result def wrapper() -> None: diff --git a/flask_app/routes/判断是否是招标文件.py b/flask_app/routes/判断是否是招标文件.py index 8437fb7..ef1aec6 100644 --- a/flask_app/routes/判断是否是招标文件.py +++ b/flask_app/routes/判断是否是招标文件.py @@ -4,7 +4,7 @@ from docx import Document import fitz from flask_app.general.llm.通义千问long import upload_file, qianwen_long -def judge_zbfile_exec(file_path,logger): +def judge_zbfile_exec(file_path): """ 判断文件是否属于招标文件,并返回结果。 """ @@ -45,7 +45,7 @@ def judge_zbfile_exec(file_path,logger): print("DOCX 文件读取失败,可能为乱码文件") return False # 解析失败直接返回 False pre_endtime=time.time() - logger.info(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") + print(f"judge_zbfile_exec预处理耗时:{pre_endtime - start_time:.2f} 秒") # 使用大模型进行判断 user_query = """该文件是否属于招标文件?如果是的话,请返回'是',如果不是的话,返回'否'。请不要返回其他解释或内容。 以下是常见的招标文件类型: @@ -59,7 +59,7 @@ def judge_zbfile_exec(file_path,logger): file_id = upload_file(file_path) model_res = qianwen_long(file_id, user_query) end_time = time.time() - logger.info(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") + print(f"judge_zbfile_exec实际耗时:{end_time - start_time:.2f} 秒") print(f"判断是否属于招标文件:{model_res} 实际耗时:{end_time - start_time:.2f} 秒") return '否' not in model_res diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 75ae428..b00343d 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -16,7 +16,7 @@ from flask_app.routes.test_preprocess import test_process_bp def create_app(): # 创建全局日志记录器 app = Flask(__name__) - app.process_pool = Pool(processes=3, maxtasksperchild=1) + app.process_pool = Pool(processes=10, maxtasksperchild=3) app.global_logger = create_logger_main('model_log') # 全局日志记录器 # 注册蓝图 app.register_blueprint(get_deviation_bp)