Update app.py
Browse files
app.py
CHANGED
@@ -72,11 +72,11 @@ def infer(prompts, negative_prompts, image):
|
|
72 |
output=np.array(output, dtype=np.float32)
|
73 |
return output
|
74 |
|
75 |
-
with gr.Blocks(css=".gradio-container {background-
|
76 |
gr.Markdown(
|
77 |
"""
|
78 |
-
<h1 style="text-align: center;">
|
79 |
-
|
80 |
</h1>
|
81 |
<h3 style="text-align: left;"> This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with a new type of conditioning.</h3>
|
82 |
<h3 style="text-align: left;"> While this is definitely a work in progress, you can still try it out by using the p5 sketch to create a keypoint image and using it as the conditioning image.</h3>
|
@@ -99,12 +99,13 @@ with gr.Blocks(css=".gradio-container {background-color: #f8d0ab};") as demo:
|
|
99 |
keypoint_tool = addp5sketch(sketch_url)
|
100 |
gr.Markdown(
|
101 |
"""
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
|
|
108 |
""")
|
109 |
|
110 |
run_btn.click(fn=infer, inputs = [prompts, negative_prompts, conditioning_image], outputs = output)
|
|
|
72 |
output=np.array(output, dtype=np.float32)
|
73 |
return output
|
74 |
|
75 |
+
with gr.Blocks(css=".gradio-container {background-image: linear-gradient(to bottom, #206dff 10%, #f8d0ab 90%)};") as demo:
|
76 |
gr.Markdown(
|
77 |
"""
|
78 |
+
<h1 style="text-align: center; font-size: 30px; color: white">
|
79 |
+
π Animal Pose Control Net π
|
80 |
</h1>
|
81 |
<h3 style="text-align: left;"> This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with a new type of conditioning.</h3>
|
82 |
<h3 style="text-align: left;"> While this is definitely a work in progress, you can still try it out by using the p5 sketch to create a keypoint image and using it as the conditioning image.</h3>
|
|
|
99 |
keypoint_tool = addp5sketch(sketch_url)
|
100 |
gr.Markdown(
|
101 |
"""
|
102 |
+
<h3 style="text-align: left;">Additional Information</h3>
|
103 |
+
<a style = "color: black; font-size: 20px" href="https://openpifpaf.github.io/plugins_animalpose.html">OpenPifPaf Animalpose</a>
|
104 |
+
<a style = "color: black; font-size: 20px" href="https://huggingface.co/datasets/JFoz/dog-cat-pose">Dataset</a>
|
105 |
+
<a style = "color: black; font-size: 20px" href="https://huggingface.co/JFoz/dog-cat-pose">Diffusers model</a>
|
106 |
+
<a style = "color: black; font-size: 20px" href="https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5"> WANDB Training Report</a>
|
107 |
+
<a style = "color: black; font-size: 20px" href="https://github.com/fi4cr/animalpose/tree/main/scripts">Training Scripts</a>
|
108 |
+
<a style = "color: black; font-size: 20px" href="https://p5js.org">p5.js</a>
|
109 |
""")
|
110 |
|
111 |
run_btn.click(fn=infer, inputs = [prompts, negative_prompts, conditioning_image], outputs = output)
|