don't import ImageGrab if it's not needed

This commit is contained in:
Maciej Budyś
2022-01-21 00:33:55 +01:00
parent 60f4ad7728
commit 71da1c6a20

View File

@@ -2,7 +2,6 @@ import time
from pathlib import Path from pathlib import Path
import PIL.Image import PIL.Image
import PIL.ImageGrab
import fire import fire
import numpy as np import numpy as np
import pyperclip import pyperclip
@@ -59,6 +58,7 @@ def run(read_from='clipboard',
mocr = MangaOcr(pretrained_model_name_or_path, force_cpu) mocr = MangaOcr(pretrained_model_name_or_path, force_cpu)
if read_from == 'clipboard': if read_from == 'clipboard':
import PIL.ImageGrab
logger.info('Reading from clipboard') logger.info('Reading from clipboard')
img = None img = None