Refactor

app.py

@@ -9,6 +9,13 @@ from huggingface_hub import hf_hub_download
 
 from model import Model
 
+DESCRIPTION = '''# StyleGAN-Human
+
+This is an unofficial demo for [https://github.com/stylegan-human/StyleGAN-Human](https://github.com/stylegan-human/StyleGAN-Human).
+'''
+FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=gradio-blocks.stylegan-human" />'
+
+
 def parse_args() -> argparse.Namespace:
     parser = argparse.ArgumentParser()
     parser.add_argument('--device', type=str, default='cpu')
@@ -25,11 +32,8 @@ def main():
     args = parse_args()
     app = Model(device=args.device)
 
-    with gr.Blocks(theme=args.theme) as demo:
-        gr.Markdown('''<center><h1>StyleGAN-Human</h1></center>
-
-This is a Blocks version of [this app](https://huggingface.co/spaces/hysts/StyleGAN-Human) and [this app](https://huggingface.co/spaces/hysts/StyleGAN-Human-Interpolation).
-''')
+    with gr.Blocks(theme=args.theme, css='style.css') as demo:
+        gr.Markdown(DESCRIPTION)
 
         with gr.Row():
             with gr.Column():
@@ -74,9 +78,7 @@ This is a Blocks version of [this app](https://huggingface.co/spaces/hysts/Style
                 with gr.Row():
                     interpolated_images = gr.Gallery(label='Output Images')
 
-        gr.Markdown(
-            '<center><img src="https://visitor-badge.glitch.me/badge?page_id=gradio-blocks.stylegan-human" alt="visitor badge"/></center>'
-        )
+        gr.Markdown(FOOTER)
 
         generate_button1.click(app.generate_single_image,
                                inputs=[seed1, psi1],

style.css

@@ -0,0 +1,7 @@
+h1 {
+  text-align: center;
+}
+img#visitor-badge {
+  display: block;
+  margin: auto;
+}