Update run/gradio_ootd.py

run/gradio_ootd.py

@@ -118,14 +118,14 @@ def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, se
  return images
 
 
-block = gr.Blocks().queue()
+block = gr.Blocks(theme="Nymbo/Alyx_Theme").queue()
 with block:
  with gr.Row():
- gr.Markdown("# OOTDiffusion Demo")
+ gr.HTML("<center><h1>OOT Diffusion Demo</h1></center>")
  with gr.Row():
- gr.Markdown("## Half-body")
+ gr.HTML("<center><h3>Half-body</h3></center>")
  with gr.Row():
- gr.Markdown("***Support upper-body garments***")
+ gr.HTML("<center><p>Support upper-body garments</p></center>")
  with gr.Row():
  with gr.Column():
  vton_img = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_hd)
@@ -184,9 +184,9 @@ with block:
 
 
  with gr.Row():
- gr.Markdown("## Full-body")
+ gr.HTML("<center><h3>Full-body</h3></center>")
  with gr.Row():
- gr.Markdown("***Support upper-body/lower-body/dresses; garment category must be paired!!!***")
+ gr.HTML("<center><p>Support upper-body/lower-body/dresses; garment category must be paired!!!</p></center>")
  with gr.Row():
  with gr.Column():
  vton_img_dc = gr.Image(label="Model", sources='upload', type="filepath", height=384, value=model_dc)