Spaces:
Runtime error
Runtime error
removed deprecation
Browse files
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.
|
| 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.
|
| 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])
|