Update app.py

app.py

@@ -13,7 +13,7 @@ class GANGenerator(torch.nn.Module):
         input_dim = latent_dim + embed_dim * 3
         
         # Output: 3 color channels * 256 * 256 image size
-        self.output_pixels = 3 * 256 * 256
+        self.output_pixels = 3 * 128 * 128
         self.fc = torch.nn.Linear(input_dim, self.output_pixels) 
         self.latent_dim = latent_dim