diff --git a/skynet/frontend/telegram.py b/skynet/frontend/telegram.py index 678e19c..a18f26a 100644 --- a/skynet/frontend/telegram.py +++ b/skynet/frontend/telegram.py @@ -103,7 +103,7 @@ async def run_skynet_telegram( else: logging.info(resp.result['id']) img_raw = base64.b64decode(bytes.fromhex(resp.result['img'])) - print(f'got image of size: {len(img_raw)}') + logging.info(f'got image of size: {len(img_raw)}') size = (512, 512) if resp.result['meta']['upscaler'] == 'x4': size = (2048, 2048) @@ -130,7 +130,12 @@ async def run_skynet_telegram( else: img_raw = base64.b64decode(bytes.fromhex(resp.result['img'])) - img = Image.frombytes('RGB', (512, 512), img_raw) + logging.info(f'got image of size: {len(img_raw)}') + size = (512, 512) + if resp.result['meta']['upscaler'] == 'x4': + size = (2048, 2048) + + img = Image.frombytes('RGB', size, img_raw) await bot.send_photo( message.chat.id,