Fix type annotation

bhky · Aug 17, 2024 · 3b9f495 · 3b9f495
1 parent d85edb2
commit 3b9f495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(),