Commit
·
a3c142f
1
Parent(s):
da83059
Update app.py
Browse files
app.py
CHANGED
@@ -320,10 +320,11 @@ def speech_to_text(video_file_path, selected_source_lang, whisper_model, num_spe
|
|
320 |
*GPU Utilization: {gpu_utilization}%, GPU Memory: {gpu_memory}MiB.*
|
321 |
"""
|
322 |
selected_whisper_model_name = selected_whisper_model.value
|
|
|
323 |
filename, _ = os.path.splitext(video_file_path)
|
324 |
filename = filename.replace(" ", "_")
|
325 |
filename = filename.replace("(", "_").replace(")", "_")
|
326 |
-
output_filename = f"{filename}_{
|
327 |
save_path = os.path.join("output", output_filename)
|
328 |
df_results = pd.DataFrame(objects)
|
329 |
df_results.to_csv(save_path)
|
|
|
320 |
*GPU Utilization: {gpu_utilization}%, GPU Memory: {gpu_memory}MiB.*
|
321 |
"""
|
322 |
selected_whisper_model_name = selected_whisper_model.value
|
323 |
+
model_name = selected_whisper_model_name.split("/")[-1]
|
324 |
filename, _ = os.path.splitext(video_file_path)
|
325 |
filename = filename.replace(" ", "_")
|
326 |
filename = filename.replace("(", "_").replace(")", "_")
|
327 |
+
output_filename = f"{filename}_{model_name}.csv"
|
328 |
save_path = os.path.join("output", output_filename)
|
329 |
df_results = pd.DataFrame(objects)
|
330 |
df_results.to_csv(save_path)
|