jhj0517 commited on
Commit
78c9bd5
1 Parent(s): 1494da1

add `initial_prompt`

Browse files
Files changed (1) hide show
  1. modules/whisper_data_class.py +8 -1
modules/whisper_data_class.py CHANGED
@@ -15,6 +15,7 @@ class WhisperGradioComponents:
15
  best_of: gr.Number
16
  patience: gr.Number
17
  condition_on_previous_text: gr.Checkbox
 
18
  """
19
  A data class to pass Gradio components to the function before Gradio pre-processing.
20
  See this documentation for more information about Gradio pre-processing: https://www.gradio.app/docs/components
@@ -52,10 +53,15 @@ class WhisperGradioComponents:
52
  patience: gr.Number
53
  Beam search patience factor.
54
 
55
- condition_on_previous_text: bool
56
  if True, the previous output of the model is provided as a prompt for the next window;
57
  disabling may make the text inconsistent across windows, but the model becomes less prone to
58
  getting stuck in a failure loop, such as repetition looping or timestamps going out of sync.
 
 
 
 
 
59
  """
60
 
61
  def to_list(self) -> list:
@@ -82,6 +88,7 @@ class WhisperValues:
82
  best_of: int
83
  patience: float
84
  condition_on_previous_text: bool
 
85
  """
86
  A data class to use Whisper parameters in your function after Gradio pre-processing.
87
  See this documentation for more information about Gradio pre-processing: : https://www.gradio.app/docs/components
 
15
  best_of: gr.Number
16
  patience: gr.Number
17
  condition_on_previous_text: gr.Checkbox
18
+ initial_prompt: gr.Textbox
19
  """
20
  A data class to pass Gradio components to the function before Gradio pre-processing.
21
  See this documentation for more information about Gradio pre-processing: https://www.gradio.app/docs/components
 
53
  patience: gr.Number
54
  Beam search patience factor.
55
 
56
+ condition_on_previous_text: gr.Checkbox
57
  if True, the previous output of the model is provided as a prompt for the next window;
58
  disabling may make the text inconsistent across windows, but the model becomes less prone to
59
  getting stuck in a failure loop, such as repetition looping or timestamps going out of sync.
60
+
61
+ initial_prompt: gr.Textbox
62
+ Optional text to provide as a prompt for the first window. This can be used to provide, or
63
+ "prompt-engineer" a context for transcription, e.g. custom vocabularies or proper nouns
64
+ to make it more likely to predict those word correctly.
65
  """
66
 
67
  def to_list(self) -> list:
 
88
  best_of: int
89
  patience: float
90
  condition_on_previous_text: bool
91
+ initial_prompt: Optional[str]
92
  """
93
  A data class to use Whisper parameters in your function after Gradio pre-processing.
94
  See this documentation for more information about Gradio pre-processing: : https://www.gradio.app/docs/components