Spaces:
Runtime error
Runtime error
ShaoTengLiu
commited on
Commit
·
7fef50a
1
Parent(s):
bb8b1f0
update two buttons
Browse files- 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
|
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 |
-
|
|
|
|
|
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,
|