Update app.py

app.py

@@ -7,7 +7,7 @@ class Generator(nn.Module):
     def __init__(self):
         super(Generator, self).__init__()
         self.main = nn.Sequential(
-            nn.ConvTranspose2d(100, 64 * 8, 4, 1, 0, bias=False),
+            nn.ConvTranspose2d(128, 64 * 8, 4, 1, 0, bias=False),
             nn.BatchNorm2d(64 * 8),
             nn.ReLU(True),
 
@@ -39,7 +39,7 @@ netG.eval()
 
 def generate_image():
     with torch.no_grad():
-        noise = torch.randn(1, 100, 1, 1)
+        noise = torch.randn(1, 128, 1, 1)
         fake_image = netG(noise)
 
     generated_image = fake_image.squeeze().cpu().numpy()