shaokun commited on
Commit
0321bd1
1 Parent(s): 384b3ad

removed deprecation

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,12 +105,12 @@ loader = ModelTrainer()
105
  with gr.Blocks() as demo:
106
  with gr.Box():
107
  instance_images = gr.Files(label='Instance images')
108
- pretrained_model_name_or_path = gr.inputs.Textbox(lines=1, label='pretrained_model_name_or_path', default='stabilityai/stable-diffusion-2-1')
109
  output_message = gr.Markdown()
110
  train_button = gr.Button('Train')
111
  train_button.click(fn=loader.train, inputs=[pretrained_model_name_or_path, instance_images], outputs=[output_message])
112
  with gr.Box():
113
- csv_input = gr.inputs.File(label='CSV File')
114
  output_message2 = gr.Markdown()
115
  generate_button = gr.Button('Generate Pictures from CSV')
116
  generate_button.click(fn=loader.generate_pictures, inputs=[csv_input], outputs=[output_message2])
 
105
  with gr.Blocks() as demo:
106
  with gr.Box():
107
  instance_images = gr.Files(label='Instance images')
108
+ pretrained_model_name_or_path = gr.Textbox(lines=1, label='pretrained_model_name_or_path', placeholder='stabilityai/stable-diffusion-2-1')
109
  output_message = gr.Markdown()
110
  train_button = gr.Button('Train')
111
  train_button.click(fn=loader.train, inputs=[pretrained_model_name_or_path, instance_images], outputs=[output_message])
112
  with gr.Box():
113
+ csv_input = gr.File(label='CSV File')
114
  output_message2 = gr.Markdown()
115
  generate_button = gr.Button('Generate Pictures from CSV')
116
  generate_button.click(fn=loader.generate_pictures, inputs=[csv_input], outputs=[output_message2])