FIX: Only send supported image types for classification (#49)

* FIX: Only send supported image types for classification
2023-04-27 17:52:20 -03:00
parent 2cd60a4b3b
commit 89ac5d720a
1 changed files with 5 additions and 1 deletions
@@ -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)