Add anime-whisper

2024-11-17 16:47:37 +09:00
parent 8d476175a7
commit 9740256c87
2 changed files with 13 additions and 9 deletions
--- a/gradio_tabs/dataset.py
+++ b/gradio_tabs/dataset.py
@@ -177,9 +177,16 @@ def create_dataset_app() -> gr.Blocks:
                    value=False,
                )
                hf_repo_id = gr.Dropdown(
-                    ["openai/whisper", "kotoba-tech/kotoba-whisper-v1.1"],
-                    label="HuggingFaceのWhisperモデル",
-                    value="openai/whisper",
+                    [
+                        "openai/whisper-large-v3-turbo",
+                        "openai/whisper-large-v3",
+                        "openai/whisper-large-v2",
+                        "kotoba-tech/kotoba-whisper-v1.1",
+                        "kotoba-tech/kotoba-whisper-v2.1",
+                        "litagin/anime-whisper",
+                    ],
+                    label="HuggingFaceのWhisper repo_id",
+                    value="openai/whisper-large-v3",
                    visible=False,
                )
                compute_type = gr.Dropdown(