Commit
·
ee7af39
1
Parent(s):
819c872
Update app.py
Browse files
app.py
CHANGED
@@ -180,7 +180,7 @@ def push(model_name, where_to_upload, hf_token):
|
|
180 |
title_instance_prompt_string = ''
|
181 |
previous_instance_prompt = instance_prompt
|
182 |
image_string = f'''{title_instance_prompt_string}
|
183 |
-
{image_string} as demo:
|
|
249 |
</div>
|
250 |
''')
|
251 |
gr.Markdown("# Dreambooth training")
|
252 |
-
gr.Markdown("Customize Stable Diffusion by giving it with few-shot examples")
|
253 |
with gr.Row():
|
254 |
type_of_thing = gr.Dropdown(label="What would you like to train?", choices=["object", "person", "style"], value="object", interactive=True)
|
255 |
|
@@ -337,4 +337,4 @@ with gr.Blocks(css=css) as demo:
|
|
337 |
generate_button.click(fn=generate, inputs=prompt, outputs=result_image)
|
338 |
push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result])
|
339 |
convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result)
|
340 |
-
demo.launch()
|
|
|
180 |
title_instance_prompt_string = ''
|
181 |
previous_instance_prompt = instance_prompt
|
182 |
image_string = f'''{title_instance_prompt_string}
|
183 |
+
{image_string}'''
|
184 |
readme_text = f'''---
|
185 |
license: creativeml-openrail-m
|
186 |
tags:
|
|
|
249 |
</div>
|
250 |
''')
|
251 |
gr.Markdown("# Dreambooth training")
|
252 |
+
gr.Markdown("Customize Stable Diffusion by giving it with few-shot examples. Based on TheLastBen's [fast-DreamBooth Colab](https://colab.research.google.com/github/TheLastBen/fast-stable-diffusion/blob/main/fast-DreamBooth.ipynb) with 🧨 diffusers")
|
253 |
with gr.Row():
|
254 |
type_of_thing = gr.Dropdown(label="What would you like to train?", choices=["object", "person", "style"], value="object", interactive=True)
|
255 |
|
|
|
337 |
generate_button.click(fn=generate, inputs=prompt, outputs=result_image)
|
338 |
push_button.click(fn=push, inputs=[model_name, where_to_upload, hf_token], outputs=[success_message_upload, result])
|
339 |
convert_button.click(fn=convert_to_ckpt, inputs=[], outputs=result)
|
340 |
+
demo.launch(debug=True)
|