discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-03-06 17:30:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* FIX: Only send supported image types for classification
This commit is contained in:
Rafael dos Santos Silva 2023-04-27 17:52:20 -03:00 committed by GitHub
parent 2cd60a4b3b
commit 89ac5d720a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/modules/nsfw/nsfw_classification.rb
View File

@ -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)