Adding example useage in whisperProgressHook
Browse files
src/hooks/whisperProgressHook.py
CHANGED
@@ -111,9 +111,21 @@ def unregister_thread_local_progress_listener(progress_listener: ProgressListene
|
|
111 |
def create_progress_listener_handle(progress_listener: ProgressListener):
|
112 |
return ProgressListenerHandle(progress_listener)
|
113 |
|
|
|
114 |
if __name__ == '__main__':
|
115 |
-
|
116 |
-
|
117 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
|
119 |
print("Done")
|
|
|
111 |
def create_progress_listener_handle(progress_listener: ProgressListener):
|
112 |
return ProgressListenerHandle(progress_listener)
|
113 |
|
114 |
+
# Example usage
|
115 |
if __name__ == '__main__':
|
116 |
+
class PrintingProgressListener:
|
117 |
+
def on_progress(self, current: Union[int, float], total: Union[int, float]):
|
118 |
+
print(f"Progress: {current}/{total}")
|
119 |
+
|
120 |
+
def on_finished(self):
|
121 |
+
print("Finished")
|
122 |
+
|
123 |
+
import whisper
|
124 |
+
model = whisper.load_model("medium")
|
125 |
+
|
126 |
+
with create_progress_listener_handle(PrintingProgressListener()) as listener:
|
127 |
+
# Set verbose to None to disable the progress bar, as we are using our own
|
128 |
+
result = model.transcribe("J:\\Dev\\OpenAI\\whisper\\tests\\Noriko\\out.mka", language="Japanese", fp16=False, verbose=None)
|
129 |
+
print(result)
|
130 |
|
131 |
print("Done")
|