Commit
•
0eddda3
1
Parent(s):
24452d6
Update app.py
Browse files
app.py
CHANGED
@@ -454,11 +454,11 @@ with gr.Blocks(css=css) as demo:
|
|
454 |
train_btn.click(fn=train, inputs=is_visible+concept_collection+file_collection+[training_summary_model_name]+[training_summary_checkbox]+[training_summary_token]+[type_of_thing]+[steps]+[perc_txt_encoder]+[swap_auto_calculated], outputs=[result, try_your_model, push_to_hub, convert_button, training_ongoing, completed_training], queue=True)
|
455 |
|
456 |
#Button to generate an image from your trained model after training
|
457 |
-
generate_button.click(fn=generate, inputs=prompt, outputs=result_image, queue=
|
458 |
#Button to push the model to the Hugging Face Hub
|
459 |
-
push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result], queue=
|
460 |
#Button to convert the model to ckpt format
|
461 |
-
convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result, queue=
|
462 |
|
463 |
#Checks if the training is running
|
464 |
demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False, show_progress=False)
|
|
|
454 |
train_btn.click(fn=train, inputs=is_visible+concept_collection+file_collection+[training_summary_model_name]+[training_summary_checkbox]+[training_summary_token]+[type_of_thing]+[steps]+[perc_txt_encoder]+[swap_auto_calculated], outputs=[result, try_your_model, push_to_hub, convert_button, training_ongoing, completed_training], queue=True)
|
455 |
|
456 |
#Button to generate an image from your trained model after training
|
457 |
+
generate_button.click(fn=generate, inputs=prompt, outputs=result_image, queue=False)
|
458 |
#Button to push the model to the Hugging Face Hub
|
459 |
+
push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result], queue=False)
|
460 |
#Button to convert the model to ckpt format
|
461 |
+
convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result, queue=False)
|
462 |
|
463 |
#Checks if the training is running
|
464 |
demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False, show_progress=False)
|