diff options
| author | Kyomotoi <[email protected]> | 2022-02-03 14:36:24 +0800 |
|---|---|---|
| committer | Kyomotoi <[email protected]> | 2022-02-03 14:36:24 +0800 |
| commit | 3e32ca3964ff8f40e0b491e87f153040f2348fd0 (patch) | |
| tree | d04d19ba5a25f6f1f6a9c4f8c398d49eb252df3c /ATRI/plugins/setu/tf_dealer.py | |
| parent | f5a020d45f7294214bbcd488955b9c391d651a6d (diff) | |
| download | ATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.tar.gz ATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.tar.bz2 ATRI-3e32ca3964ff8f40e0b491e87f153040f2348fd0.zip | |
🔖 更新版本:
更新记录请参考文档: atri.kyomotoi.moe/changelog/overview/
Diffstat (limited to 'ATRI/plugins/setu/tf_dealer.py')
| -rw-r--r-- | ATRI/plugins/setu/tf_dealer.py | 10 |
1 files changed, 3 insertions, 7 deletions
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()) |