Spaces:
Running
Running
jhj0517
commited on
Commit
·
fbc1421
1
Parent(s):
32ef602
debug
Browse files- app.py +1 -1
- modules/whisper/whisper_base.py +43 -43
app.py
CHANGED
@@ -160,7 +160,7 @@ class App:
|
|
160 |
diarization_device=dd_diarization_device)
|
161 |
|
162 |
btn_run.click(fn=self.whisper_inf.transcribe_file,
|
163 |
-
inputs=params
|
164 |
outputs=[tb_indicator, files_subtitles])
|
165 |
dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
166 |
|
|
|
160 |
diarization_device=dd_diarization_device)
|
161 |
|
162 |
btn_run.click(fn=self.whisper_inf.transcribe_file,
|
163 |
+
inputs=params,
|
164 |
outputs=[tb_indicator, files_subtitles])
|
165 |
dd_model.change(fn=self.on_change_models, inputs=[dd_model], outputs=[cb_translate])
|
166 |
|
modules/whisper/whisper_base.py
CHANGED
@@ -131,7 +131,7 @@ class WhisperBase(ABC):
|
|
131 |
file_format: str,
|
132 |
add_timestamp: bool,
|
133 |
progress=gr.Progress(),
|
134 |
-
|
135 |
):
|
136 |
"""
|
137 |
Write subtitle file from Files
|
@@ -157,48 +157,48 @@ class WhisperBase(ABC):
|
|
157 |
Output file path to return to gr.Files()
|
158 |
"""
|
159 |
print("transcrption started: ")
|
160 |
-
try:
|
161 |
-
|
162 |
-
|
163 |
-
|
164 |
-
|
165 |
-
|
166 |
-
|
167 |
-
|
168 |
-
|
169 |
-
|
170 |
-
|
171 |
-
|
172 |
-
|
173 |
-
|
174 |
-
|
175 |
-
|
176 |
-
|
177 |
-
|
178 |
-
|
179 |
-
|
180 |
-
|
181 |
-
|
182 |
-
|
183 |
-
|
184 |
-
|
185 |
-
|
186 |
-
|
187 |
-
|
188 |
-
|
189 |
-
|
190 |
-
|
191 |
-
|
192 |
-
|
193 |
-
|
194 |
-
|
195 |
-
|
196 |
-
except Exception as e:
|
197 |
-
|
198 |
-
finally:
|
199 |
-
|
200 |
-
|
201 |
-
|
202 |
|
203 |
@spaces.GPU(duration=120)
|
204 |
def transcribe_mic(self,
|
|
|
131 |
file_format: str,
|
132 |
add_timestamp: bool,
|
133 |
progress=gr.Progress(),
|
134 |
+
#*whisper_params,
|
135 |
):
|
136 |
"""
|
137 |
Write subtitle file from Files
|
|
|
157 |
Output file path to return to gr.Files()
|
158 |
"""
|
159 |
print("transcrption started: ")
|
160 |
+
# try:
|
161 |
+
# files_info = {}
|
162 |
+
# for file in files:
|
163 |
+
# print("run started: ")
|
164 |
+
# transcribed_segments, time_for_task = self.run(
|
165 |
+
# file.name,
|
166 |
+
# progress,
|
167 |
+
# *whisper_params,
|
168 |
+
# )
|
169 |
+
# print("run finished: ")
|
170 |
+
#
|
171 |
+
# file_name, file_ext = os.path.splitext(os.path.basename(file.name))
|
172 |
+
# file_name = safe_filename(file_name)
|
173 |
+
# subtitle, file_path = self.generate_and_write_file(
|
174 |
+
# file_name=file_name,
|
175 |
+
# transcribed_segments=transcribed_segments,
|
176 |
+
# add_timestamp=add_timestamp,
|
177 |
+
# file_format=file_format,
|
178 |
+
# output_dir=self.output_dir
|
179 |
+
# )
|
180 |
+
# print("generated sub finished: ")
|
181 |
+
# files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path}
|
182 |
+
#
|
183 |
+
# total_result = ''
|
184 |
+
# total_time = 0
|
185 |
+
# for file_name, info in files_info.items():
|
186 |
+
# total_result += '------------------------------------\n'
|
187 |
+
# total_result += f'{file_name}\n\n'
|
188 |
+
# total_result += f'{info["subtitle"]}'
|
189 |
+
# total_time += info["time_for_task"]
|
190 |
+
#
|
191 |
+
# result_str = f"Done in {self.format_time(total_time)}! Subtitle is in the outputs folder.\n\n{total_result}"
|
192 |
+
# result_file_path = [info['path'] for info in files_info.values()]
|
193 |
+
#
|
194 |
+
# return [result_str, result_file_path]
|
195 |
+
#
|
196 |
+
# except Exception as e:
|
197 |
+
# print(f"Error transcribing file: {e}")
|
198 |
+
# finally:
|
199 |
+
# self.release_cuda_memory()
|
200 |
+
# if not files:
|
201 |
+
# self.remove_input_files([file.name for file in files])
|
202 |
|
203 |
@spaces.GPU(duration=120)
|
204 |
def transcribe_mic(self,
|