Update app.py

app.py

@@ -338,7 +338,7 @@ class App:
                     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_dataframe2,inputs=tb_indicator,outputs=tb_indicator)
+                    btn_run.click(fn=self.update_dataframe,inputs=[cb_timestamp_preview,whisper_params["is_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])
@@ -400,9 +400,6 @@ class App:
         else:
             return gr.Dataframe(headers=["Text"], column_widths=["100%"])
 
-    @staticmethod
-    def update_dataframe2(input_value):
-            return gr.Dataframe(headers=["Time","Text"], column_widths=["15%","85%"])
 
 # Create the parser for command-line arguments
 parser = argparse.ArgumentParser()