Update app.py

app.py

@@ -19,7 +19,7 @@ pretransform = transforms.Compose([
 def pipeline(img):
     img = model_build.img_preprocess(img, pretransform)
     with torch.no_grad():
-        _, latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)
+        latent_space = psp(img.float(), randomize_noise=True, resize=False, return_latents=True)
     img = stylegan2(latent_space, noise_mode='none')
     img = Image.fromarray(np.array((img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8).squeeze(0)[20:-20,:,:]))
     img.save('output.png')