Update app.py

Update background color and description

app.py

@@ -66,14 +66,15 @@ def infer(prompts, negative_prompts, image):
     output=np.array(output, dtype=np.float32)
     return output
 
-with gr.Blocks(theme='kfahn/AnimalPose') as demo:  
+with gr.Blocks(css=".gradio-container {background-color: #45f10c};") as demo:  
   gr.Markdown(
       """
       <h1 style="text-align: center;">
-      Animal Pose Control Net
+      🐕 Animal Pose Control Net 🐈
       </h1>
-     
-      <h3 style="text-align: center;"> This is a demo of Animal Pose ControlNet, which is a model trained on runwayml/stable-diffusion-v1-5 with new type of conditioning.
+      <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>  
+      <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>
+      <h3 style="text-align: left;"> The model was generated as part of the Hugging Face Jax Diffusers sprint.  Thank you to both Hugging Face and Google Cloud who provided the TPUs for training!
       </h3>
       """) 
   with gr.Row():
@@ -82,7 +83,7 @@ with gr.Blocks(theme='kfahn/AnimalPose') as demo:
       negative_prompts  = gr.Textbox(label="Negative Prompt", value="lowres, two heads, bad muzzle, bad anatomy, missing ears, missing paws")
       conditioning_image = gr.Image(label="Conditioning Image")
       run_btn = gr.Button("Run")
-      wandb = wandb_report(report_url)
+      #wandb = wandb_report(report_url)
     with gr.Column():
       keypoint_tool = addp5sketch(sketch_url)
       output = gr.Image(
@@ -93,7 +94,7 @@ with gr.Blocks(theme='kfahn/AnimalPose') as demo:
         [Dataset](https://huggingface.co/datasets/JFoz/dog-poses-controlnet-dataset)  
         [Diffusers model](https://huggingface.co/JFoz/dog-pose)  
         [Github](https://github.com/fi4cr/animalpose)   
-        [Training Report](https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5)
+        [WANDB Training Report](https://wandb.ai/john-fozard/dog-cat-pose/runs/kmwcvae5)
         """)     
 
   run_btn.click(fn=infer, inputs = [prompts, negative_prompts, conditioning_image], outputs = output)