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
import PIL.Image
import PIL.ImageGrab
import fire
import numpy as np
import pyperclip
@@ -59,6 +58,7 @@ def run(read_from='clipboard',
mocr = MangaOcr(pretrained_model_name_or_path, force_cpu)
if read_from == 'clipboard':
import PIL.ImageGrab
logger.info('Reading from clipboard')
img = None