Update UI

app.py

@@ -73,13 +73,10 @@ def infer(
 
 
 ### Description
-header = r"""
-<h1>ID-Booth: Identity-consistent Face Generation with Diffusion Models</h1>
-Official Gradio demo for <a href='https://dariant.github.io/publications/ID-Booth' target='_blank'>ID-Booth: Identity-consistent Face Generation with Diffusion Models</a>
-"""
+header = " # ID-Booth: Identity-consistent Face Generation with Diffusion Models"
+description = "This is an official Gradio demo for the paper <a href='https://dariant.github.io/publications/ID-Booth' target='_blank'>ID-Booth: Identity-consistent Face Generation with Diffusion Models</a>"
 
 footer = r"""
----
 📝 **Citation**
 <br>
 If you find ID-Booth helpful, please consider citing our paper:
@@ -93,20 +90,18 @@ If you find ID-Booth helpful, please consider citing our paper:
 ```
 """
 
-css = """
-#col-container {
-    margin: 0 auto;
-    max-width: 960px;
-}
-"""
+css = '''
+.gradio-container {width: 60% !important}
+'''
 
 with gr.Blocks(css=css) as demo:
 
     # description
     gr.Markdown(header)
+    gr.Markdown(description)
     with gr.Column(elem_id="col-container"):
-        with gr.Row():
-            gr.Markdown("### Choose an identity, background, and pose:")
+        # with gr.Row():
+        #     gr.Markdown("### Choose an identity, background, and pose:")
 
         with gr.Row():
             for id in id_list:
@@ -139,7 +134,7 @@ with gr.Blocks(css=css) as demo:
                 value=poses_list[0],
             )
 
-        run_button = gr.Button("Generate", scale=0, variant="primary")
+        run_button = gr.Button("Generate in-the-wild images", scale=0, variant="primary")
         #result = gr.Image(label="Result", show_label=False)
         result = gr.Gallery(label="Generated Images", show_label=False)
 
@@ -152,7 +147,7 @@ with gr.Blocks(css=css) as demo:
                 value="cartoon, cgi, render, illustration, painting, drawing, black and white, bad body proportions, landscape",
             )
             num_inference_steps = gr.Slider( 
-                label="Number of sample steps",
+                label="Number of sampling steps",
                 minimum=1,
                 maximum=100,
                 step=1,