Update app.py
Browse filesfix array errors
app.py
CHANGED
@@ -26,6 +26,8 @@ control_img = 'myimage.jpg'
|
|
26 |
|
27 |
examples = [["a yellow dog standing in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
28 |
|
|
|
|
|
29 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
30 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
31 |
)
|
@@ -85,7 +87,7 @@ with gr.Blocks(css=".gradio-container {background-color: #45f10c};") as demo:
|
|
85 |
with gr.Column():
|
86 |
prompts = gr.Textbox(label="Prompt", placeholder="animal standing, best quality, highres")
|
87 |
negative_prompts = gr.Textbox(label="Negative Prompt", value="lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws")
|
88 |
-
conditioning_image = gr.Image(label="Conditioning Image", value=default_example[
|
89 |
run_btn = gr.Button("Run")
|
90 |
#wandb = wandb_report(report_url)
|
91 |
with gr.Column():
|
|
|
26 |
|
27 |
examples = [["a yellow dog standing in grass", "lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws", "example1.jpg"]]
|
28 |
|
29 |
+
default_example = examples[0]
|
30 |
+
|
31 |
controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
|
32 |
"JFoz/dog-cat-pose", dtype=jnp.bfloat16
|
33 |
)
|
|
|
87 |
with gr.Column():
|
88 |
prompts = gr.Textbox(label="Prompt", placeholder="animal standing, best quality, highres")
|
89 |
negative_prompts = gr.Textbox(label="Negative Prompt", value="lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws")
|
90 |
+
conditioning_image = gr.Image(label="Conditioning Image", value=default_example[3])
|
91 |
run_btn = gr.Button("Run")
|
92 |
#wandb = wandb_report(report_url)
|
93 |
with gr.Column():
|