From 89ac5d720a736a8e8cb752dc2eb5dd0b37d854b7 Mon Sep 17 00:00:00 2001 From: Rafael dos Santos Silva Date: Thu, 27 Apr 2023 17:52:20 -0300 Subject: [PATCH] FIX: Only send supported image types for classification (#49) * FIX: Only send supported image types for classification --- lib/modules/nsfw/nsfw_classification.rb | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lib/modules/nsfw/nsfw_classification.rb b/lib/modules/nsfw/nsfw_classification.rb index de10db5c..a21a3a86 100644 --- a/lib/modules/nsfw/nsfw_classification.rb +++ b/lib/modules/nsfw/nsfw_classification.rb @@ -65,7 +65,11 @@ module DiscourseAi end def content_of(target_to_classify) - target_to_classify.uploads.to_a.select { |u| FileHelper.is_supported_image?(u.url) } + target_to_classify + .uploads + .where(extension: %w[png jpeg jpg PNG JPEG JPG]) + .to_a + .select { |u| FileHelper.is_supported_image?(u.url) } end def opennsfw2_verdict?(clasification)