Update app.py

app.py

@@ -12,6 +12,7 @@ import gradio as gr
 
 from types import SimpleNamespace
 
+#Edited from run_generator.py to return PIL images instead of saving them to the disk.
 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')
@@ -83,13 +84,13 @@ def inference(seed):
     return generate_images(
         G, 
         SimpleNamespace(**{
-            'truncation_psi': 0.7, #It seems like 0.7 will give the best result.
+            'truncation_psi': 0.7, #It seems like 0.7 will give the best result for this model.
             'seeds': [seed],
             'batch_size': 1,
             'pixel_min': -1,
             'pixel_max': 1,
         }) #https://github.com/adriansahlman/stylegan2_pytorch/blob/master/run_generator.py
-    )
+    )[0]
     
 title = "TWDNEv3 CPU Generator"
 description = "Gradio Demo for TWDNEv3 CPU Generator (stylegan2_pytorch port). To use it, simply put your random seed."