multimodalart HF staff commited on
Commit
480e9b0
1 Parent(s): ee0769d

Update app_training.py

Browse files
Files changed (1) hide show
  1. app_training.py +8 -5
app_training.py CHANGED
@@ -25,17 +25,20 @@ def create_training_demo(trainer: Trainer,
25
  max_lines=1,
26
  placeholder='A man is surfing')
27
  gr.Markdown('''
28
- - Upload a video and write a prompt describing the video.
29
  ''')
30
  with gr.Box():
31
  gr.Markdown('Output Model')
32
  output_model_name = gr.Text(label='Name of your model',
 
33
  max_lines=1)
34
  delete_existing_model = gr.Checkbox(
35
  label='Delete existing model of the same name',
36
- value=False,
37
- visible=False)
38
- validation_prompt = gr.Text(label='Validation Prompt')
 
 
39
 
40
  with gr.Column():
41
  with gr.Box():
@@ -74,7 +77,7 @@ def create_training_demo(trainer: Trainer,
74
  value=100,
75
  precision=0)
76
  gr.Markdown('''
77
- - The base model must be a model that is compatible with [diffusers](https://github.com/huggingface/diffusers) library.
78
  - Expected time to train a model for 300 steps: ~20 minutes with T4
79
  - You can check the training status by pressing the "Open logs" button if you are running this on your Space.
80
  ''')
 
25
  max_lines=1,
26
  placeholder='A man is surfing')
27
  gr.Markdown('''
28
+ - Upload a video and write a prompt that describes the video.
29
  ''')
30
  with gr.Box():
31
  gr.Markdown('Output Model')
32
  output_model_name = gr.Text(label='Name of your model',
33
+ placeholder='The surfer man'
34
  max_lines=1)
35
  delete_existing_model = gr.Checkbox(
36
  label='Delete existing model of the same name',
37
+ value=False)
38
+ validation_prompt = gr.Text(label='Validation Prompt', placeholder='a dog is surfing')
39
+ gr.Markdown('''
40
+ - The validation prompt should modify your `Training Prompt` in order to validate the model
41
+ ''')
42
 
43
  with gr.Column():
44
  with gr.Box():
 
77
  value=100,
78
  precision=0)
79
  gr.Markdown('''
80
+ - The base model must be a Stable Diffusion model compatible with [diffusers](https://github.com/huggingface/diffusers) library.
81
  - Expected time to train a model for 300 steps: ~20 minutes with T4
82
  - You can check the training status by pressing the "Open logs" button if you are running this on your Space.
83
  ''')