Done?

app.py

@@ -9,6 +9,8 @@ import stylegan2
 from stylegan2 import utils
 from huggingface_hub import hf_hub_download
 
+from types import SimpleNamespace
+
 def generate_images(G, args):
     latent_size, label_size = G.latent_size, G.label_size
     device = torch.device(args.gpu[0] if args.gpu else 'cpu')
@@ -77,6 +79,16 @@ def generate_images(G, args):
 def inference(seed):
     G = stylegan2.models.load(hf_hub_download("hr16/Gwern-TWDNEv3-pytorch_ckpt", "Gs.pth", use_auth_token=os.environ['MODEL_READING_TOKEN']))
     G.eval()
+    return generate_images(
+        G, 
+        SimpleNamespace(**{
+            'truncation_psi': 0.7, #It seems like 0.7 will give the best result.
+            'seeds': [seed],
+            'batch_size': 1,
+            'pixel_min': -1,
+            'pixel_max': 1,
+        }) #https://github.com/adriansahlman/stylegan2_pytorch/blob/master/run_generator.py
+    )
     
 title = "TWDNEv3 CPU Generator"
 description = "Gradio Demo for TWDNEv3 CPU Generator (stylegan2_pytorch port). To use it, simply put your random seed."