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)