From 8c95f0ff75ecbbc21f36dd8cdee64244a4983df7 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Thu, 14 Mar 2024 16:45:31 +0900
Subject: [PATCH] Fix hf whisper

---
 colab.ipynb   | 4 +++-
 transcribe.py | 4 ----
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/colab.ipynb b/colab.ipynb
index 0684088..5e80f52 100644
--- a/colab.ipynb
+++ b/colab.ipynb
@@ -26,7 +26,9 @@
       "source": [
         "## 0. 環境構築\n",
         "\n",
-        "Style-Bert-VITS2の環境をcolab上に構築します。グラボモードが有効になっていることを確認し、以下のセルを順に実行してください。"
+        "Style-Bert-VITS2の環境をcolab上に構築します。グラボモードが有効になっていることを確認し、以下のセルを順に実行してください。\n",
+        "\n",
+        "最近のcolabのアップデートにより、エラーダイアログ「WARNING: The following packages were previously imported in this runtime: [pydevd_plugins]」が出るが、「キャンセル」を選択して続行してください。"
       ]
     },
     {
diff --git a/transcribe.py b/transcribe.py
index 582a89a..f039226 100644
--- a/transcribe.py
+++ b/transcribe.py
@@ -84,10 +84,6 @@ def transcribe_files_with_hf_whisper(
 
     results: list[str] = []
     for whisper_result in pipe(dataset):
-        logger.debug(whisper_result)
-        for result in enumerate(whisper_result):
-            logger.debug(result)
-            logger.debug(f"Transcribed: {result['text']}")
         text: str = whisper_result["text"]
         # なぜかテキストの最初に" {initial_prompt}"が入るので、文字の最初からこれを削除する
         # cf. https://github.com/huggingface/transformers/issues/27594