Update app.py

app.py

@@ -276,7 +276,8 @@ class App:
             cb_translate_output,
             dd_translate_model,
             dd_target_lang,
-            cb_timestamp_preview
+            cb_timestamp_preview,
+            cb_diarize
         )
 
     def launch(self):
@@ -333,12 +334,12 @@ class App:
                             files_subtitles = gr.Files(label="Output data", interactive=False, file_count="multiple")
                             # btn_openfolder = gr.Button('📂', scale=1)                        
 
-                    params = [input_file_audio, input_file_video, input_file_multi, input_multi, tb_input_folder, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview]
+                    params = [input_file_audio, input_file_video, input_file_multi, input_multi, tb_input_folder, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview, cb_diarize]
  
                     btn_run.click(fn=self.whisper_inf.transcribe_file,
                                   inputs=params + whisper_params.as_list(),
                                   outputs=[tb_indicator, files_subtitles, tb_info])
-                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,whisper_params["is_diarize"]],outputs=tb_indicator)
+                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,cb_diarize,outputs=tb_indicator)
                     # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
 
                     input_multi.change(fn=self.update_viewer,inputs=input_multi,outputs=[input_file_audio,input_file_video,input_file_multi])