ShaoTengLiu commited on
Commit
7fef50a
1 Parent(s): bb8b1f0

update two buttons

Browse files
Files changed (1) hide show
  1. app_training.py +4 -2
app_training.py CHANGED
@@ -122,7 +122,7 @@ def create_training_demo(trainer: Trainer,
122
  value=False,
123
  interactive=bool(os.getenv('SPACE_ID')),
124
  visible=False)
125
- run_button = gr.Button('Start Training')
126
 
127
  with gr.Box():
128
  gr.Markdown('Output message')
@@ -142,7 +142,9 @@ def create_training_demo(trainer: Trainer,
142
  remove_gpu_after_training, input_token, blend_word_1, blend_word_2, eq_params_1, eq_params_2
143
  ],
144
  outputs=output_message)
145
- run_button.click(
 
 
146
  fn=trainer.run_p2p,
147
  inputs=[
148
  training_video, training_prompt, output_model_name,
 
122
  value=False,
123
  interactive=bool(os.getenv('SPACE_ID')),
124
  visible=False)
125
+ run_button = gr.Button('Start Tuning')
126
 
127
  with gr.Box():
128
  gr.Markdown('Output message')
 
142
  remove_gpu_after_training, input_token, blend_word_1, blend_word_2, eq_params_1, eq_params_2
143
  ],
144
  outputs=output_message)
145
+
146
+ run_button_p2p = gr.Button('Start P2P')
147
+ run_button_p2p.click(
148
  fn=trainer.run_p2p,
149
  inputs=[
150
  training_video, training_prompt, output_model_name,