From 3e32ca3964ff8f40e0b491e87f153040f2348fd0 Mon Sep 17 00:00:00 2001
From: Kyomotoi <kyomotoiowo@gmail.com>
Date: Thu, 3 Feb 2022 14:36:24 +0800
Subject: =?UTF-8?q?=F0=9F=94=96=20=E6=9B=B4=E6=96=B0=E7=89=88=E6=9C=AC:?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
更新记录请参考文档: atri.kyomotoi.moe/changelog/overview/
---
ATRI/plugins/setu/tf_dealer.py | 10 +++-------
1 file changed, 3 insertions(+), 7 deletions(-)
(limited to 'ATRI/plugins/setu/tf_dealer.py')
diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py
index bf68020..f966636 100644
--- a/ATRI/plugins/setu/tf_dealer.py
+++ b/ATRI/plugins/setu/tf_dealer.py
@@ -1,7 +1,6 @@
import io
import os
import re
-import string
import asyncio
import skimage
import skimage.io
@@ -9,12 +8,8 @@ import numpy as np
from PIL import Image
from pathlib import Path
from sys import getsizeof
-from random import sample
-try:
- import tflite_runtime.interpreter as tf # type: ignore
-except Exception:
- import tensorflow as tf
+import tensorflow as tf
from ATRI.log import logger as log
from ATRI.utils import request
@@ -116,4 +111,5 @@ async def init_module():
raise WriteError("NSFW TF module init failed!")
-asyncio.get_event_loop().run_until_complete(init_module())
+loop = asyncio.get_event_loop()
+loop.create_task(init_module())
--
cgit v1.2.3