Upload app.py

app.py

@@ -68,11 +68,11 @@ def generate_grid():
     args = cfg.parse_args()
 
     with torch.no_grad():
-        # z = torch.randn(N_SAMPLES, latent_dim, device=device)
-        z = torch.cuda.FloatTensor(truncnorm.rvs(-1, 1, loc=0, scale=1, size=(N_SAMPLES, args.latent_dim)))
+        z_np = truncnorm.rvs(-1, 1, loc=0, scale=1, size=(N_SAMPLES, latent_dim)).astype(np.float32)
+        z = torch.from_numpy(z_np).to(device)
         imgs = G(z)  # (B,3,H,W) in [-1,1] or [0,1]
 
-        # make a 10x10 grid; normalize handles [-1,1]
+        # make a grid; normalize handles [-1,1]
         grid = make_grid(imgs, nrow=N_COLS, normalize=True, scale_each=True)
         return _to_pil_grid(grid)