jbilcke-hf HF Staff commited on
Commit
e4a3a7d
·
1 Parent(s): 9a2a7a2

fix gradio event chain

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -61,36 +61,38 @@ class VideoTrainerUI:
61
  self.log_parser = TrainingLogParser()
62
 
63
  def update_captioning_buttons_start(self):
64
- return {
65
- "run_autocaption_btn": gr.Button(
 
66
  interactive=False,
67
  variant="secondary",
68
  ),
69
- "stop_autocaption_btn": gr.Button(
70
  interactive=True,
71
  variant="stop",
72
  ),
73
- "copy_files_to_training_dir_btn": gr.Button(
74
  interactive=False,
75
  variant="secondary",
76
  )
77
- }
78
 
79
  def update_captioning_buttons_end(self):
80
- return {
81
- "run_autocaption_btn": gr.Button(
 
82
  interactive=True,
83
  variant="primary",
84
  ),
85
- "stop_autocaption_btn": gr.Button(
86
  interactive=False,
87
  variant="secondary",
88
  ),
89
- "copy_files_to_training_dir_btn": gr.Button(
90
  interactive=True,
91
  variant="primary",
92
  )
93
- }
94
 
95
  def show_refreshing_status(self) -> List[List[str]]:
96
  """Show a 'Refreshing...' status in the dataframe"""
 
61
  self.log_parser = TrainingLogParser()
62
 
63
  def update_captioning_buttons_start(self):
64
+ """Return individual button values instead of a dictionary"""
65
+ return (
66
+ gr.Button(
67
  interactive=False,
68
  variant="secondary",
69
  ),
70
+ gr.Button(
71
  interactive=True,
72
  variant="stop",
73
  ),
74
+ gr.Button(
75
  interactive=False,
76
  variant="secondary",
77
  )
78
+ )
79
 
80
  def update_captioning_buttons_end(self):
81
+ """Return individual button values instead of a dictionary"""
82
+ return (
83
+ gr.Button(
84
  interactive=True,
85
  variant="primary",
86
  ),
87
+ gr.Button(
88
  interactive=False,
89
  variant="secondary",
90
  ),
91
+ gr.Button(
92
  interactive=True,
93
  variant="primary",
94
  )
95
+ )
96
 
97
  def show_refreshing_status(self) -> List[List[str]]:
98
  """Show a 'Refreshing...' status in the dataframe"""