TrialAccountHF commited on
Commit
a3c142f
·
1 Parent(s): da83059

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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}_{selected_whisper_model_name}.csv"
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)