Update

README.md

@@ -4,9 +4,12 @@ emoji: 🏢
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 3.19.1
+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

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import functools
-import os
 import pickle
 import sys
 
@@ -16,18 +15,11 @@ from huggingface_hub import hf_hub_download
 sys.path.insert(0, 'StyleGAN-Human')
 
 TITLE = 'StyleGAN-Human (Interpolation)'
-DESCRIPTION = '''This is an unofficial demo for https://github.com/stylegan-human/StyleGAN-Human.
-
-Related App: [StyleGAN-Human](https://huggingface.co/spaces/hysts/StyleGAN-Human)
-'''
-
-HF_TOKEN = os.getenv('HF_TOKEN')
+DESCRIPTION = 'https://github.com/stylegan-human/StyleGAN-Human'
 
 
 def load_model(file_name: str, device: torch.device) -> 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()
@@ -74,12 +66,12 @@ def generate_interpolated_images(seed0: int, psi0: float, seed1: int,
 
 device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
 model = load_model('stylegan_human_v2_1024.pkl', device)
-func = functools.partial(generate_interpolated_images,
-                         model=model,
-                         device=device)
+fn = functools.partial(generate_interpolated_images,
+                       model=model,
+                       device=device)
 
 gr.Interface(
-    fn=func,
+    fn=fn,
     inputs=[
         gr.Slider(label='Seed 1',
                   minimum=0,
@@ -112,4 +104,4 @@ gr.Interface(
     outputs=gr.Gallery(label='Output Images', type='numpy'),
     title=TITLE,
     description=DESCRIPTION,
-).launch(show_api=False)
+).queue(max_size=10).launch()

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