multimodalart HF staff commited on
Commit
148af15
1 Parent(s): 214d6ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -442,16 +442,16 @@ with gr.Blocks(css=css) as demo:
442
  train_btn.click(lambda:gr.update(visible=True), inputs=None, outputs=training_ongoing)
443
 
444
  #The main train function
445
- 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])
446
 
447
  #Button to generate an image from your trained model after training
448
- generate_button.click(fn=generate, inputs=prompt, outputs=result_image)
449
  #Button to push the model to the Hugging Face Hub
450
- push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result])
451
  #Button to convert the model to ckpt format
452
- convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result)
453
 
454
  #Checks if the training is running
455
  demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False)
456
 
457
- demo.launch(debug=True)
 
442
  train_btn.click(lambda:gr.update(visible=True), inputs=None, outputs=training_ongoing)
443
 
444
  #The main train function
445
+ 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)
446
 
447
  #Button to generate an image from your trained model after training
448
+ generate_button.click(fn=generate, inputs=prompt, outputs=result_image, queue=True)
449
  #Button to push the model to the Hugging Face Hub
450
+ push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result], queue=True)
451
  #Button to convert the model to ckpt format
452
+ convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result, queue=True)
453
 
454
  #Checks if the training is running
455
  demo.load(fn=check_status, inputs=top_description, outputs=[top_description, try_your_model, push_to_hub, result, convert_button], queue=False)
456
 
457
+ demo.queue(default_enabled=False).launch(debug=True)