multimodalart HF staff commited on
Commit
0eddda3
1 Parent(s): 24452d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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=True)
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=True)
460
  #Button to convert the model to ckpt format
461
- convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result, queue=True)
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)