From 3e08be541e2a1ca72c8f9551077554db52a015bc Mon Sep 17 00:00:00 2001
From: Bosco Yung <15840328+bhky@users.noreply.github.com>
Date: Sun, 18 Aug 2024 07:56:32 +0900
Subject: [PATCH] Fix type annotations

---
 opennsfw2/_inference.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/opennsfw2/_inference.py b/opennsfw2/_inference.py
index 241bd64..bd69e73 100644
--- a/opennsfw2/_inference.py
+++ b/opennsfw2/_inference.py
@@ -35,7 +35,7 @@ def _load_pil_image(image_handle: Union[str, Image.Image]) -> Image.Image:
 
 
 def predict_image(
-        image_handle: str,
+        image_handle: Union[str, Image.Image],
         preprocessing: Preprocessing = Preprocessing.YAHOO,
         weights_path: Optional[str] = get_default_weights_path(),
         grad_cam_path: Optional[str] = None,
@@ -64,7 +64,7 @@ def predict_image(
 
 def _predict_from_image_handles_in_batches(
         model_: Model,
-        image_handles: Sequence[str],
+        image_handles: Union[Sequence[str], Sequence[Image.Image]],
         batch_size: int,
         preprocessing: Preprocessing
 ) -> NDFloat32Array: