Update configs creation in Google Colab notebook

ieasybooks · Jul 1, 2024 · 1020e81 · 1020e81
1 parent 42b9973
commit 1020e81
Showing 1 changed file with 27 additions and 18 deletions.
diff --git a/colab_notebook.ipynb b/colab_notebook.ipynb
@@ -134,24 +134,33 @@
         "    print('جارٍ تحويل المواد إلى نصوص باستخدام نماذج Whisper.')\n",
         "\n",
         "config = Config(\n",
-        "    urls_or_paths=list(map(str.strip, urls.split(' '))) if len(urls.strip()) else ['.'],\n",
-        "    skip_if_output_exist=False,\n",
-        "    playlist_items='',\n",
-        "    verbose=False,\n",
-        "    model_name_or_path=model,\n",
-        "    task='transcribe',\n",
-        "    language=language,\n",
-        "    use_faster_whisper=True,\n",
-        "    beam_size=5,\n",
-        "    ct2_compute_type='default',\n",
-        "    wit_client_access_tokens=[wit_api_key],\n",
-        "    max_cutting_duration=max_cutting_duration,\n",
-        "    min_words_per_segment=min_words_per_segment,\n",
-        "    save_files_before_compact=False,\n",
-        "    save_yt_dlp_responses=False,\n",
-        "    output_sample=0,\n",
-        "    output_formats=['txt', 'srt'],\n",
-        "    output_dir=output_dir,\n",
+        "    input=Config.Input(\n",
+        "        urls_or_paths=list(map(str.strip, urls.split(' '))) if len(urls.strip()) else ['.'],\n",
+        "        skip_if_output_exist=False,\n",
+        "        playlist_items='',\n",
+        "        download_retries=3,\n",
+        "        verbose=False,\n",
+        "    ),\n",
+        "    whisper=Config.Whisper(\n",
+        "        model_name_or_path=model,\n",
+        "        task='transcribe',\n",
+        "        language=language,\n",
+        "        use_faster_whisper=True,\n",
+        "        beam_size=5,\n",
+        "        ct2_compute_type='default',\n",
+        "    ),\n",
+        "    wit=Config.Wit(\n",
+        "        wit_client_access_tokens=[wit_api_key],\n",
+        "        max_cutting_duration=max_cutting_duration,\n",
+        "    ),\n",
+        "    output=Config.Output(\n",
+        "        min_words_per_segment=min_words_per_segment,\n",
+        "        save_files_before_compact=False,\n",
+        "        save_yt_dlp_responses=False,\n",
+        "        output_sample=0,\n",
+        "        output_formats=['txt', 'srt'],\n",
+        "        output_dir=output_dir,\n",
+        "    )\n",
         ")\n",
         "\n",
         "deque(farrigh(config), maxlen=0)\n",