summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-06-09 22:32:12 +0800
committerKyomotoi <[email protected]>2022-06-09 22:32:12 +0800
commit88a4105e7cbfbcd7952b3d47be3279f021f7ad6f (patch)
tree4ea6c61f0c058ae6eef57678ed3717d7d4ffe6bf /ATRI/plugins/setu
parenta8163fc8b97612793d3b55ebd4e90c61cd4ba77a (diff)
downloadATRI-88a4105e7cbfbcd7952b3d47be3279f021f7ad6f.tar.gz
ATRI-88a4105e7cbfbcd7952b3d47be3279f021f7ad6f.tar.bz2
ATRI-88a4105e7cbfbcd7952b3d47be3279f021f7ad6f.zip
💩 优化代码
Diffstat (limited to 'ATRI/plugins/setu')
-rw-r--r--ATRI/plugins/setu/tf_dealer.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py
index 58b4337..a6bd3f0 100644
--- a/ATRI/plugins/setu/tf_dealer.py
+++ b/ATRI/plugins/setu/tf_dealer.py
@@ -1,5 +1,4 @@
import io
-import os
import re
import asyncio
import skimage
@@ -18,8 +17,8 @@ from ATRI.exceptions import RequestError, WriteFileError
SETU_PATH = Path(".") / "data" / "database" / "setu"
TEMP_PATH = Path(".") / "data" / "temp"
-os.makedirs(SETU_PATH, exist_ok=True)
-os.makedirs(TEMP_PATH, exist_ok=True)
+SETU_PATH.mkdir(parents=True, exist_ok=True)
+TEMP_PATH.mkdir(parents=True, exist_ok=True)
MODULE_URL = "https://cdn.jsdelivr.net/gh/Kyomotoi/CDN@master/project/ATRI/nsfw.tflite"