From b334766f41a14bc82c0ff7af58d3bb976ca0a10a Mon Sep 17 00:00:00 2001
From: Kacper Krasowiak <krasowiak@google.com>
Date: Fri, 25 Oct 2024 09:53:30 +0000
Subject: [PATCH] fix: Aligned the flag naming convention with the argument
 naming convention.

Change-Id: I8a284d4cfdbd3f301b2c527b791ad77ed1619d0a
---
 main.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 92e4bb1..4fe7dbd 100644
--- a/main.py
+++ b/main.py
@@ -178,8 +178,8 @@
     40,
     "Top-k sampling parameter.",
 )
-_GEMINI_SAFETY_SETTINGS = flags.DEFINE_string(
-    "gemini_safety_settings",
+_SAFETY_SETTINGS = flags.DEFINE_string(
+    "_safety_settings",
     "Medium",
     "The indicator of what kind of Gemini safety settings should"
     " be used in the dubbing process. Can be"
@@ -253,7 +253,7 @@ def main(argv: Sequence[str]) -> None:
       top_p=_TOP_P.value,
       top_k=_TOP_K.value,
       max_output_tokens=_MAX_OUTPUT_TOKENS.value,
-      safety_settings=get_safety_settings(_GEMINI_SAFETY_SETTINGS.value),
+      safety_settings=get_safety_settings(_SAFETY_SETTINGS.value),
       use_elevenlabs=False if _ELEVENLABS_TOKEN.value == "".strip() else True,
       elevenlabs_token=_ELEVENLABS_TOKEN.value,
       elevenlabs_clone_voices=_ELEVENLABS_CLONE_VOICES.value,