diff options
Diffstat (limited to 'ATRI/plugins/setu/tf_dealer.py')
| -rw-r--r-- | ATRI/plugins/setu/tf_dealer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py index f41de49..bf68020 100644 --- a/ATRI/plugins/setu/tf_dealer.py +++ b/ATRI/plugins/setu/tf_dealer.py @@ -49,14 +49,14 @@ def prepare_image(img): return image -async def detect_image(url) -> list: +async def detect_image(url: str, file_size: int) -> list: try: req = await request.get(url) except RequestError: raise RequestError("Get info from download image failed!") img_byte = getsizeof(req.read()) // 1024 - if img_byte < 256: + if img_byte < file_size: return [0, 0] try: |