Update

.pre-commit-config.yaml

@@ -20,26 +20,17 @@ repos:
   - id: docformatter
     args: ['--in-place']
 - repo: https://github.com/pycqa/isort
-  rev: 5.10.1
+  rev: 5.12.0
   hooks:
     - id: isort
 - repo: https://github.com/pre-commit/mirrors-mypy
-  rev: v0.812
+  rev: v0.991
   hooks:
     - id: mypy
       args: ['--ignore-missing-imports']
+      additional_dependencies: ['types-python-slugify']
 - repo: https://github.com/google/yapf
   rev: v0.32.0
   hooks:
   - id: yapf
     args: ['--parallel', '--in-place']
-- repo: https://github.com/kynan/nbstripout
-  rev: 0.5.0
-  hooks:
-    - id: nbstripout
-      args: ['--extra-keys', 'metadata.interpreter metadata.kernelspec cell.metadata.pycharm']
-- repo: https://github.com/nbQA-dev/nbQA
-  rev: 1.3.1
-  hooks:
-    - id: nbqa-isort
-    - id: nbqa-yapf

README.md

@@ -4,9 +4,12 @@ emoji: 🌍
 colorFrom: pink
 colorTo: red
 sdk: gradio
-sdk_version: 3.0.17
+sdk_version: 3.35.2
 app_file: app.py
 pinned: false
+suggested_hardware: t4-small
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+
+https://arxiv.org/abs/2204.11823

app.py

@@ -2,111 +2,80 @@
 
 from __future__ import annotations
 
-import argparse
-
 import gradio as gr
 
 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')
-    parser.add_argument('--theme', type=str)
-    parser.add_argument('--share', action='store_true')
-    parser.add_argument('--port', type=int)
-    parser.add_argument('--disable-queue',
-                        dest='enable_queue',
-                        action='store_false')
-    return parser.parse_args()
-
-
-def main():
-    args = parse_args()
-    model = Model(device=args.device)
-
-    with gr.Blocks(theme=args.theme, css='style.css') as demo:
-        gr.Markdown(DESCRIPTION)
-
-        with gr.Row():
-            with gr.Column():
-                with gr.Row():
-                    seed1 = gr.Number(value=6876, label='Seed 1')
-                    psi1 = gr.Slider(0,
-                                     2,
-                                     value=0.7,
-                                     step=0.05,
-                                     label='Truncation psi 1')
-                with gr.Row():
-                    generate_button1 = gr.Button('Generate')
-                with gr.Row():
-                    generated_image1 = gr.Image(type='numpy',
-                                                label='Generated Image 1')
-
-            with gr.Column():
-                with gr.Row():
-                    seed2 = gr.Number(value=6886, label='Seed 2')
-                    psi2 = gr.Slider(0,
-                                     2,
-                                     value=0.7,
-                                     step=0.05,
-                                     label='Truncation psi 2')
-                with gr.Row():
-                    generate_button2 = gr.Button('Generate')
-                with gr.Row():
-                    generated_image2 = gr.Image(type='numpy',
-                                                label='Generated Image 2')
-
-        with gr.Row():
-            with gr.Column():
-                with gr.Row():
-                    num_frames = gr.Slider(
-                        0,
-                        41,
-                        value=7,
-                        step=1,
-                        label='Number of Intermediate Frames')
-                with gr.Row():
-                    interpolate_button = gr.Button('Interpolate')
-                with gr.Row():
-                    interpolated_images = gr.Gallery(label='Output Images')
-
-        gr.Markdown(FOOTER)
-
-        generate_button1.click(model.generate_single_image,
-                               inputs=[
-                                   seed1,
-                                   psi1,
-                               ],
-                               outputs=generated_image1)
-        generate_button2.click(model.generate_single_image,
-                               inputs=[
-                                   seed2,
-                                   psi2,
-                               ],
-                               outputs=generated_image2)
-        interpolate_button.click(model.generate_interpolated_images,
-                                 inputs=[
-                                     seed1,
-                                     psi1,
-                                     seed2,
-                                     psi2,
-                                     num_frames,
-                                 ],
-                                 outputs=interpolated_images)
-
-    demo.launch(
-        enable_queue=args.enable_queue,
-        server_port=args.port,
-        share=args.share,
-    )
-
-
-if __name__ == '__main__':
-    main()
+DESCRIPTION = '# [StyleGAN-Human](https://github.com/stylegan-human/StyleGAN-Human)'
+
+model = Model()
+
+with gr.Blocks(css='style.css') as demo:
+    gr.Markdown(DESCRIPTION)
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                seed1 = gr.Number(label='Seed 1', value=6876)
+                psi1 = gr.Slider(label='Truncation psi 1',
+                                 minimum=0,
+                                 maximum=2,
+                                 step=0.05,
+                                 value=0.7)
+            with gr.Row():
+                generate_button1 = gr.Button('Generate')
+            with gr.Row():
+                generated_image1 = gr.Image(label='Generated Image 1',
+                                            type='numpy',
+                                            height=600)
+
+        with gr.Column():
+            with gr.Row():
+                seed2 = gr.Number(label='Seed 2', value=6886)
+                psi2 = gr.Slider(label='Truncation psi 2',
+                                 minimum=0,
+                                 maximum=2,
+                                 step=0.05,
+                                 value=0.7)
+            with gr.Row():
+                generate_button2 = gr.Button('Generate')
+            with gr.Row():
+                generated_image2 = gr.Image(label='Generated Image 2',
+                                            type='numpy',
+                                            height=600)
+
+    with gr.Row():
+        with gr.Column():
+            with gr.Row():
+                num_frames = gr.Slider(label='Number of Intermediate Frames',
+                                       minimum=0,
+                                       maximum=41,
+                                       step=1,
+                                       value=7)
+            with gr.Row():
+                interpolate_button = gr.Button('Interpolate')
+            with gr.Row():
+                interpolated_images = gr.Gallery(label='Output Images',
+                                                 object_fit='scale-down')
+
+    generate_button1.click(model.generate_single_image,
+                           inputs=[
+                               seed1,
+                               psi1,
+                           ],
+                           outputs=generated_image1)
+    generate_button2.click(model.generate_single_image,
+                           inputs=[
+                               seed2,
+                               psi2,
+                           ],
+                           outputs=generated_image2)
+    interpolate_button.click(model.generate_interpolated_images,
+                             inputs=[
+                                 seed1,
+                                 psi1,
+                                 seed2,
+                                 psi2,
+                                 num_frames,
+                             ],
+                             outputs=interpolated_images)
+demo.queue(max_size=10).launch()

model.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import os
 import pathlib
 import pickle
 import sys
@@ -14,18 +13,16 @@ app_dir = pathlib.Path(__file__).parent
 submodule_dir = app_dir / 'StyleGAN-Human'
 sys.path.insert(0, submodule_dir.as_posix())
 
-HF_TOKEN = os.environ['HF_TOKEN']
-
 
 class Model:
-    def __init__(self, device: str | torch.device):
-        self.device = torch.device(device)
+    def __init__(self):
+        self.device = torch.device(
+            'cuda:0' if torch.cuda.is_available() else 'cpu')
         self.model = self.load_model('stylegan_human_v2_1024.pkl')
 
     def load_model(self, file_name: str) -> nn.Module:
-        path = hf_hub_download('hysts/StyleGAN-Human',
-                               f'models/{file_name}',
-                               use_auth_token=HF_TOKEN)
+        path = hf_hub_download('public-data/StyleGAN-Human',
+                               f'models/{file_name}')
         with open(path, 'rb') as f:
             model = pickle.load(f)['G_ema']
         model.eval()

requirements.txt

@@ -1,5 +1,5 @@
-numpy==1.22.3
-Pillow==9.1.0
-scipy==1.8.0
-torch==1.11.0
-torchvision==0.12.0
+numpy==1.23.5
+Pillow==10.0.0
+scipy==1.10.1
+torch==2.0.1
+torchvision==0.15.2