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])
|