fffiloni commited on
Commit
e719472
1 Parent(s): 2dce22c
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -137,8 +137,18 @@ def process_image(input_pil, remove_bg):
137
  torch.cuda.empty_cache()
138
  return output_video[0]
139
 
 
 
 
 
 
 
 
 
 
140
  def gradio_interface():
141
- with gr.Blocks() as app:
 
142
  gr.Markdown("# PSHuman: Photorealistic Single-image 3D Human Reconstruction using Cross-Scale Multiview Diffusion and Explicit Remeshing")
143
  gr.HTML("""
144
  <div style="display:flex;column-gap:4px;">
@@ -174,7 +184,7 @@ def gradio_interface():
174
 
175
  submit_button = gr.Button("Process")
176
 
177
- output_video= gr.Video(label="Output Video", scale=4)
178
 
179
  gr.Examples(
180
  examples = examples_folder,
 
137
  torch.cuda.empty_cache()
138
  return output_video[0]
139
 
140
+ css="""
141
+ div#col-container{
142
+ margin: 0 auto;
143
+ max-width: 982px;
144
+ }
145
+ div#video-out-elm{
146
+ height: 323px;
147
+ }
148
+ """
149
  def gradio_interface():
150
+ with gr.Blocks(css=css) as app:
151
+ with gr.Column(elem_id="col-container"):
152
  gr.Markdown("# PSHuman: Photorealistic Single-image 3D Human Reconstruction using Cross-Scale Multiview Diffusion and Explicit Remeshing")
153
  gr.HTML("""
154
  <div style="display:flex;column-gap:4px;">
 
184
 
185
  submit_button = gr.Button("Process")
186
 
187
+ output_video= gr.Video(label="Output Video", scale=4, elem_id="video-out-elm")
188
 
189
  gr.Examples(
190
  examples = examples_folder,