Spaces:
Sleeping
Sleeping
When the transcribe whisper language is set to Chinese,
Browse filesthe initial prompt is automatically added "繁體: ",
and the initial prompt mode is set to "prepend_all_segments"
app.py
CHANGED
|
@@ -102,10 +102,14 @@ class WhisperTranscriber:
|
|
| 102 |
vad, vadMergeWindow, vadMaxMergeSize,
|
| 103 |
word_timestamps: bool = False, highlight_words: bool = False,
|
| 104 |
progress=gr.Progress()):
|
| 105 |
-
|
|
|
|
|
|
|
|
|
|
| 106 |
vadOptions = VadOptions(vad, vadMergeWindow, vadMaxMergeSize, self.app_config.vad_padding, self.app_config.vad_prompt_window, self.app_config.vad_initial_prompt_mode)
|
| 107 |
|
| 108 |
return self.transcribe_webui(modelName, languageName, urlData, multipleFiles, microphoneData, task, vadOptions,
|
|
|
|
| 109 |
word_timestamps=word_timestamps, highlight_words=highlight_words, progress=progress)
|
| 110 |
|
| 111 |
# Entry function for the full tab
|
|
@@ -140,6 +144,10 @@ class WhisperTranscriber:
|
|
| 140 |
else:
|
| 141 |
temperature = [temperature]
|
| 142 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 143 |
vadOptions = VadOptions(vad, vadMergeWindow, vadMaxMergeSize, vadPadding, vadPromptWindow, vadInitialPromptMode)
|
| 144 |
|
| 145 |
return self.transcribe_webui(modelName, languageName, urlData, multipleFiles, microphoneData, task, vadOptions,
|
|
|
|
| 102 |
vad, vadMergeWindow, vadMaxMergeSize,
|
| 103 |
word_timestamps: bool = False, highlight_words: bool = False,
|
| 104 |
progress=gr.Progress()):
|
| 105 |
+
if languageName == "Chinese":
|
| 106 |
+
initial_prompt = "繁體: "
|
| 107 |
+
self.app_config.vad_initial_prompt_mode = "prepend_all_segments"
|
| 108 |
+
|
| 109 |
vadOptions = VadOptions(vad, vadMergeWindow, vadMaxMergeSize, self.app_config.vad_padding, self.app_config.vad_prompt_window, self.app_config.vad_initial_prompt_mode)
|
| 110 |
|
| 111 |
return self.transcribe_webui(modelName, languageName, urlData, multipleFiles, microphoneData, task, vadOptions,
|
| 112 |
+
initial_prompt=initial_prompt,
|
| 113 |
word_timestamps=word_timestamps, highlight_words=highlight_words, progress=progress)
|
| 114 |
|
| 115 |
# Entry function for the full tab
|
|
|
|
| 144 |
else:
|
| 145 |
temperature = [temperature]
|
| 146 |
|
| 147 |
+
if languageName == "Chinese":
|
| 148 |
+
initial_prompt = "繁體: " + initial_prompt
|
| 149 |
+
self.app_config.vad_initial_prompt_mode = "prepend_all_segments"
|
| 150 |
+
|
| 151 |
vadOptions = VadOptions(vad, vadMergeWindow, vadMaxMergeSize, vadPadding, vadPromptWindow, vadInitialPromptMode)
|
| 152 |
|
| 153 |
return self.transcribe_webui(modelName, languageName, urlData, multipleFiles, microphoneData, task, vadOptions,
|