Spaces:
Runtime error
Runtime error
Set default initial prompt mode
Browse files
app.py
CHANGED
|
@@ -273,15 +273,21 @@ class WhisperTranscriber:
|
|
| 273 |
if ('task' in decodeOptions):
|
| 274 |
task = decodeOptions.pop('task')
|
| 275 |
|
| 276 |
-
|
| 277 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 278 |
# Prepend initial prompt
|
| 279 |
-
prompt_strategy = PrependPromptStrategy(initial_prompt,
|
| 280 |
elif (vadOptions.vadInitialPromptMode == VadInitialPromptMode.JSON_PROMPT_MODE):
|
| 281 |
# Use a JSON format to specify the prompt for each segment
|
| 282 |
prompt_strategy = JsonPromptStrategy(initial_prompt)
|
| 283 |
else:
|
| 284 |
-
raise ValueError("Invalid vadInitialPromptMode: " +
|
| 285 |
|
| 286 |
# Callable for processing an audio file
|
| 287 |
whisperCallable = model.create_callback(language, task, prompt_strategy=prompt_strategy, **decodeOptions)
|
|
|
|
| 273 |
if ('task' in decodeOptions):
|
| 274 |
task = decodeOptions.pop('task')
|
| 275 |
|
| 276 |
+
initial_prompt_mode = vadOptions.vadInitialPromptMode
|
| 277 |
+
|
| 278 |
+
# Set default initial prompt mode
|
| 279 |
+
if (initial_prompt_mode is None):
|
| 280 |
+
initial_prompt_mode = VadInitialPromptMode.PREPREND_FIRST_SEGMENT
|
| 281 |
+
|
| 282 |
+
if (initial_prompt_mode == VadInitialPromptMode.PREPEND_ALL_SEGMENTS or
|
| 283 |
+
initial_prompt_mode == VadInitialPromptMode.PREPREND_FIRST_SEGMENT):
|
| 284 |
# Prepend initial prompt
|
| 285 |
+
prompt_strategy = PrependPromptStrategy(initial_prompt, initial_prompt_mode)
|
| 286 |
elif (vadOptions.vadInitialPromptMode == VadInitialPromptMode.JSON_PROMPT_MODE):
|
| 287 |
# Use a JSON format to specify the prompt for each segment
|
| 288 |
prompt_strategy = JsonPromptStrategy(initial_prompt)
|
| 289 |
else:
|
| 290 |
+
raise ValueError("Invalid vadInitialPromptMode: " + initial_prompt_mode)
|
| 291 |
|
| 292 |
# Callable for processing an audio file
|
| 293 |
whisperCallable = model.create_callback(language, task, prompt_strategy=prompt_strategy, **decodeOptions)
|