Update app.py

app.py

@@ -3,7 +3,7 @@ import keras_cv
 import gradio as gr
 
 
-## load model
+# load keras model
 resolution = 512
 dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True, 
@@ -29,8 +29,9 @@ def generate_images(prompt: str, num_imgs_to_gen: int, num_steps: int):
    
     return generated_img
     
+css_code='.gradio-container {background-color:lightskyblue} .container {background: lightskyblue !important}'
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=css_code) as demo:
     gr.HTML("<h2 style=\"font-size: 2em; font-weight: bold\" align=\"center\">Keras Dreambooth - Pink Floyd Division Bell Demo</h2>")
     
     gr.Markdown("This model has been fine tuned to learn the concept of Division Bell from Pink Floyd's famous album `The Division Bell`")