From 6f71a3abb7dc26bd41c9c555eae31048caaec6c3 Mon Sep 17 00:00:00 2001 From: Bosco Yung <15840328+bhky@users.noreply.github.com> Date: Sun, 18 Aug 2024 08:20:44 +0900 Subject: [PATCH] Small rearrangement --- opennsfw2/_inference.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opennsfw2/_inference.py b/opennsfw2/_inference.py index 745e555..97cdb9b 100644 --- a/opennsfw2/_inference.py +++ b/opennsfw2/_inference.py @@ -108,9 +108,9 @@ def predict_images( from ._inspection import make_and_save_nsfw_grad_cam for image_handle, grad_cam_path in zip(image_handles, grad_cam_paths): + pil_image = _load_pil_image(image_handle) make_and_save_nsfw_grad_cam( - _load_pil_image(image_handle), preprocessing, global_model, - grad_cam_path, alpha + pil_image, preprocessing, global_model, grad_cam_path, alpha ) return nsfw_probabilities