From b84fb40d1dbde7922c789094d06cebcd52fd6061 Mon Sep 17 00:00:00 2001 From: bhaskarkishore <50350646+bhaskarkishore@users.noreply.github.com> Date: Wed, 28 Feb 2024 15:01:47 +0530 Subject: [PATCH] Refactored rekognition normalization for clarity --- src/aws/mod.rs | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/aws/mod.rs b/src/aws/mod.rs index ce0f473..a62eb3d 100644 --- 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,