Update app_inference.py

app_inference.py

@@ -33,6 +33,12 @@ body {
     font-size: 250%;
 }
 
+.lora-title {
+    text-align: center;
+    border-radius: 10px;
+    background: linear-gradient(90deg, #1CB5E0 0%, #000851 100%);
+}
+
 .gr-image {
     width: 512px;
     height: 512px;
@@ -157,14 +163,14 @@ with gr.Blocks(css=css) as demo:
     with gr.Row(elem_classes="gr-row"):
         with gr.Column():
             with gr.Group(elem_classes="lora-column"):
-                gr.Markdown('## Content B-LoRA')
+                content_sub_title = gr.HTML('''<h2>Content B-LoRA</h2>''', elem_classes="lora-title")
                 content_checkbox = gr.Checkbox(label='Use Content Only', value=False)
                 content_lora_model_id = gr.Dropdown(label='Model ID', choices=[])
                 content_prompt = gr.Text(label='Content instance prompt', interactive=False, max_lines=1)
                 content_image = gr.Image(label='Content Image', elem_classes="gr-image")
         with gr.Column():
             with gr.Group(elem_classes="lora-column"):
-                gr.Markdown('## Style B-LoRA')
+                style_sub_title = gr.HTML('''<h2>Style B-LoRA</h2>''', elem_classes="lora-title")
                 style_checkbox = gr.Checkbox(label='Use Style Only', value=False)
                 style_lora_model_id = gr.Dropdown(label='Model ID', choices=[])
                 style_prompt = gr.Text(label='Style instance prompt', interactive=False, max_lines=1)