Update colab

ionic-bond · Apr 28, 2024 · c7bd337 · c7bd337
1 parent 0f5008f
commit c7bd337
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/stream_translator.ipynb b/stream_translator.ipynb
@@ -29,6 +29,7 @@
       },
       "outputs": [],
       "source": [
+        "# @title Install\n",
         "!apt install libcublas11\n",
         "!pip install stream-translator-gpt"
       ]
@@ -41,9 +42,18 @@
       },
       "outputs": [],
       "source": [
-        "!stream-translator-gpt https://www.youtube.com/watch?v=r1dytEjbUqo --model large --language ja \\\n",
+        "# @title Run\n",
+        "URL = \"https://www.youtube.com/watch?v=r1dytEjbUqo\" # @param {type:\"string\"}\n",
+        "model = \"large\" # @param [\"small\", \"medium\", \"large\"] {allow-input: true}\n",
+        "language = \"ja\" # @param {type:\"string\"}\n",
+        "translation_prompt = \"Translate from Japanese to Chinese, only output the translation, do not output the original text\" # @param {type:\"string\"}\n",
+        "translation_engine = \"google\" # @param [\"openai\", \"google\"]\n",
+        "api_key = \"AIzaSyCPYq1F3WOlSGbWFcERf5qCv1hlo2SKOG8\" # @param {type:\"string\"}\n",
+        "translation_history_size = \"2\" # @param [\"0\", \"1\", \"2\", \"3\", \"4\", \"5\"] {allow-input: true}\n",
+        "\n",
+        "!stream-translator-gpt \"{URL}\" --model {model} --language {language} \\\n",
         "  --use_faster_whisper \\\n",
-        "  --gpt_translation_prompt \"Translate from Japanese to Chinese, only output the translation, do not output the original text\" --google_api_key \"AIzaSyCPYq1F3WOlSGbWFcERf5qCv1hlo2SKOG8\" --gpt_translation_history_size 2"
+        "  --gpt_translation_prompt \"{translation_prompt}\" --{translation_engine}_api_key \"{api_key}\" --gpt_translation_history_size {translation_history_size}"
       ]
     }
   ],