Update app.py

app.py

@@ -57,13 +57,13 @@ class Generator(nn.Module):
 
  # More downsampling
  model3 = []
- out_features = in_features*3
+ out_features = in_features//2
  for _ in range(2):
  model3 += [ nn.ConvTranspose2d(in_features, out_features, 3, stride=2, padding=1, output_padding=1),
  norm_layer(out_features),
  nn.ReLU(inplace=True) ]
  in_features = out_features
- out_features = in_features*3
+ out_features = in_features//2
  self.model3 = nn.Sequential(*model3)
 
  # Output layer