update

app.py

@@ -50,6 +50,7 @@ def gen(md, model_name, md2, text, seed, nb_samples, width, height):
             cond[1][1:] = cond[1][0:1]
             
         x_init = torch.randn(nb_samples, 3, height, width).to(device)
+        print(x_init.shape)
         fake_sample = sample(model, x_init=x_init, cond=cond)
         fake_sample = (fake_sample + 1) / 2
     grid = torchvision.utils.make_grid(fake_sample, nrow=4)