Update app_inference.py

app_inference.py

@@ -12,16 +12,6 @@ from huggingface_hub import HfApi
 from inf import InferencePipeline
 
 
-TITLE = '# Implicit Style-Content Separation using B-LoRA'
-
-
-DESC = '''<center>
-    This is a demo for our <a href="https://arxiv.org/abs/2403.14572">paper</a>: <b>''Implicit Style-Content Separation using B-LoRA''</b>.
-    <br>
-    Project page and code is available <a href="https://b-lora.github.io/B-LoRA/">here</a>.
-</center>'''
-
-
 SAMPLE_MODEL_IDS = [
     'lora-library/B-LoRA-teddybear',
     'lora-library/B-LoRA-bull',
@@ -35,8 +25,12 @@ body {
     font-size: 30px;
 }
 
-h1 {
-  text-align: center;
+#title {
+    text-align: center;
+}
+
+#title h1 {
+    font-size: 250%;
 }
 
 .gr-image {
@@ -150,6 +144,14 @@ app = InferenceUtil(hf_token)
 
 
 with gr.Blocks(css=css) as demo:
+    title = gr.HTML(
+        '''<h1>Implicit Style-Content Separation using B-LoRA</h1>
+        <p>This is a demo for our <a href="https://arxiv.org/abs/2403.14572">paper</a>: <b>''Implicit Style-Content Separation using B-LoRA''</b>.
+    <br>
+    Project page and code is available <a href="https://b-lora.github.io/B-LoRA/">here</a>.</p>
+        ''',
+        elem_id="title"
+      )
     gr.Markdown(TITLE)
     gr.Markdown(DESC)
     with gr.Row(elem_classes="gr-row"):