Commit
·
18fd9e7
1
Parent(s):
0d04ccc
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def predict(Seed,noise_mode,truncation_psi,trans_x,trans_y,angle):
|
|
| 58 |
img = G(z, label, truncation_psi=truncation_psi, noise_mode=noise_mode)
|
| 59 |
img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
|
| 60 |
|
| 61 |
-
return PIL.Image.fromarray(img[0].cpu().numpy()[:,:,0])
|
| 62 |
|
| 63 |
|
| 64 |
|
|
|
|
| 58 |
img = G(z, label, truncation_psi=truncation_psi, noise_mode=noise_mode)
|
| 59 |
img = (img.permute(0, 2, 3, 1) * 127.5 + 128).clamp(0, 255).to(torch.uint8)
|
| 60 |
|
| 61 |
+
return (PIL.Image.fromarray(img[0].cpu().numpy()[:,:,0])).resize((512,512))
|
| 62 |
|
| 63 |
|
| 64 |
|