Update app.py

app.py

@@ -58,7 +58,7 @@ def predict(Seed,noise_mode,truncation_psi,trans_x,trans_y,angle):
     img = G(z, label, truncation_psi=truncation_psi, noise_mode=noise_mode)
     img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
       
-    return PIL.Image.fromarray(img[0].cpu().numpy()[:,:,0])
+    return (PIL.Image.fromarray(img[0].cpu().numpy()[:,:,0])).resize((512,512))