Refactored rekognition normalization for clarity

Cryptonomic · Feb 28, 2024 · b84fb40 · b84fb40
1 parent db9c57f
commit b84fb40
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/aws/mod.rs b/src/aws/mod.rs
@@ -90,15 +90,16 @@ impl Rekognition {
         match input {
             "Explicit Nudity" => ModerationCategories::ExplicitNudity,
             "Suggestive" => ModerationCategories::Suggestive,
-            "Violence" => ModerationCategories::Violence,
-            "Visually Disturbing" => ModerationCategories::VisuallyDisturbing,
-            "Rude" => ModerationCategories::Rude,
             "Drugs" => ModerationCategories::Drugs,
             "Tobacco" => ModerationCategories::Tobacco,
-            "Alcohol" => ModerationCategories::Alcohol,
-            "Gambling" => ModerationCategories::Gambling,
+            "Rude" => ModerationCategories::Rude,
             "Hate" => ModerationCategories::Hate,
             // Model 7
+            "Violence" => ModerationCategories::Violence,
+            "Visually Disturbing" => ModerationCategories::VisuallyDisturbing,
+            "Alcohol" => ModerationCategories::Alcohol,
+            "Gambling" => ModerationCategories::Gambling,
+            "Hate Symbols" => ModerationCategories::Hate,
             "Explicit" => ModerationCategories::ExplicitContent,
             "Rude Gestures" => ModerationCategories::Rude,
             "Drugs & Tobacco" => ModerationCategories::DrugsAndTobacco,