kfahn commited on
Commit
a9b5cc4
β€’
1 Parent(s): 58d4aa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
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-color: #f8d0ab};") as demo:
76
  gr.Markdown(
77
  """
78
- <h1 style="text-align: center;">
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,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
- [OpenPifPaf Animalpose](https://openpifpaf.github.io/plugins_animalpose.html).
103
- [Dataset](https://huggingface.co/datasets/JFoz/dog-cat-pose)
104
- [Diffusers model](https://huggingface.co/JFoz/dog-cat-pose)
105
- [WANDB Training Report](https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5)
106
- [Training Scripts](https://github.com/fi4cr/animalpose/tree/main/scripts)
107
- [p5.js](https://p5js.org)
 
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)