From 3b9f495d147fa0c5225034a039a4b324aa8b8901 Mon Sep 17 00:00:00 2001 From: Bosco Yung <15840328+bhky@users.noreply.github.com> Date: Sun, 18 Aug 2024 08:03:24 +0900 Subject: [PATCH] Fix type annotation --- opennsfw2/_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opennsfw2/_inference.py b/opennsfw2/_inference.py index bd69e73..745e555 100644 --- a/opennsfw2/_inference.py +++ b/opennsfw2/_inference.py @@ -86,7 +86,7 @@ def _predict_from_image_handles_in_batches( def predict_images( - image_handles: Sequence[str], + image_handles: Union[Sequence[str], Sequence[Image.Image]], batch_size: int = 8, preprocessing: Preprocessing = Preprocessing.YAHOO, weights_path: Optional[str] = get_default_weights_path(),