From 6f0d21ac1d412d9d4d63ec9519750a210c2f8194 Mon Sep 17 00:00:00 2001 From: Bosco Yung <15840328+bhky@users.noreply.github.com> Date: Sat, 17 Aug 2024 16:25:47 +0900 Subject: [PATCH] Fix var names --- opennsfw2/_inference.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/opennsfw2/_inference.py b/opennsfw2/_inference.py index 7a4a34f..646666f 100644 --- a/opennsfw2/_inference.py +++ b/opennsfw2/_inference.py @@ -75,10 +75,10 @@ def _predict_from_image_handles_in_batches( """ prediction_batches: List[Any] = [] for i in range(0, len(image_handles), batch_size): - path_batch = image_handles[i: i + batch_size] + handle_batch = image_handles[i: i + batch_size] image_batch = [ - preprocess_image(_load_pil_image(path), preprocessing) - for path in path_batch + preprocess_image(_load_pil_image(handle), preprocessing) + for handle in handle_batch ] prediction_batches.append(model_(np.array(image_batch))) predictions: NDFloat32Array = np.concatenate(prediction_batches, axis=0)