DarianT commited on
Commit
88ca7f6
·
1 Parent(s): b482274
Files changed (1) hide show
  1. app.py +10 -15
app.py CHANGED
@@ -73,13 +73,10 @@ def infer(
73
 
74
 
75
  ### Description
76
- header = r"""
77
- <h1>ID-Booth: Identity-consistent Face Generation with Diffusion Models</h1>
78
- 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>
79
- """
80
 
81
  footer = r"""
82
- ---
83
  📝 **Citation**
84
  <br>
85
  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:
93
  ```
94
  """
95
 
96
- css = """
97
- #col-container {
98
- margin: 0 auto;
99
- max-width: 960px;
100
- }
101
- """
102
 
103
  with gr.Blocks(css=css) as demo:
104
 
105
  # description
106
  gr.Markdown(header)
 
107
  with gr.Column(elem_id="col-container"):
108
- with gr.Row():
109
- gr.Markdown("### Choose an identity, background, and pose:")
110
 
111
  with gr.Row():
112
  for id in id_list:
@@ -139,7 +134,7 @@ with gr.Blocks(css=css) as demo:
139
  value=poses_list[0],
140
  )
141
 
142
- run_button = gr.Button("Generate", scale=0, variant="primary")
143
  #result = gr.Image(label="Result", show_label=False)
144
  result = gr.Gallery(label="Generated Images", show_label=False)
145
 
@@ -152,7 +147,7 @@ with gr.Blocks(css=css) as demo:
152
  value="cartoon, cgi, render, illustration, painting, drawing, black and white, bad body proportions, landscape",
153
  )
154
  num_inference_steps = gr.Slider(
155
- label="Number of sample steps",
156
  minimum=1,
157
  maximum=100,
158
  step=1,
 
73
 
74
 
75
  ### Description
76
+ header = " # ID-Booth: Identity-consistent Face Generation with Diffusion Models"
77
+ 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>"
 
 
78
 
79
  footer = r"""
 
80
  📝 **Citation**
81
  <br>
82
  If you find ID-Booth helpful, please consider citing our paper:
 
90
  ```
91
  """
92
 
93
+ css = '''
94
+ .gradio-container {width: 60% !important}
95
+ '''
 
 
 
96
 
97
  with gr.Blocks(css=css) as demo:
98
 
99
  # description
100
  gr.Markdown(header)
101
+ gr.Markdown(description)
102
  with gr.Column(elem_id="col-container"):
103
+ # with gr.Row():
104
+ # gr.Markdown("### Choose an identity, background, and pose:")
105
 
106
  with gr.Row():
107
  for id in id_list:
 
134
  value=poses_list[0],
135
  )
136
 
137
+ run_button = gr.Button("Generate in-the-wild images", scale=0, variant="primary")
138
  #result = gr.Image(label="Result", show_label=False)
139
  result = gr.Gallery(label="Generated Images", show_label=False)
140
 
 
147
  value="cartoon, cgi, render, illustration, painting, drawing, black and white, bad body proportions, landscape",
148
  )
149
  num_inference_steps = gr.Slider(
150
+ label="Number of sampling steps",
151
  minimum=1,
152
  maximum=100,
153
  step=1,