From 5c911c71a18e325863b0ef7af78ba04cab78ade2 Mon Sep 17 00:00:00 2001 From: Kyomotoi Date: Sat, 14 May 2022 00:00:26 +0800 Subject: =?UTF-8?q?=E2=99=BF=EF=B8=8F=20=E4=BC=98=E5=8C=96=E6=8A=A5?= =?UTF-8?q?=E9=94=99=E4=BF=A1=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ATRI/plugins/setu/tf_dealer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'ATRI/plugins/setu') diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py index f966636..58b4337 100644 --- a/ATRI/plugins/setu/tf_dealer.py +++ b/ATRI/plugins/setu/tf_dealer.py @@ -13,7 +13,7 @@ import tensorflow as tf from ATRI.log import logger as log from ATRI.utils import request -from ATRI.exceptions import RequestError, WriteError +from ATRI.exceptions import RequestError, WriteFileError SETU_PATH = Path(".") / "data" / "database" / "setu" @@ -60,8 +60,8 @@ async def detect_image(url: str, file_size: int) -> list: path = TEMP_PATH / f"{file_name}.jpg" with open(path, "wb") as f: f.write(req.read()) - except WriteError: - raise WriteError("Writing file failed!") + except WriteFileError: + raise WriteFileError("Writing file failed!") await init_module() model_path = str((SETU_PATH / "nsfw.tflite").absolute()) @@ -107,8 +107,8 @@ async def init_module(): with open(path, "wb") as w: w.write(data.read()) log.info("模型装载完成") - except WriteError: - raise WriteError("NSFW TF module init failed!") + except WriteFileError: + raise WriteFileError("NSFW TF module init failed!") loop = asyncio.get_event_loop() -- cgit v1.2.3