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