fixed issues with features

app.py

@@ -34,15 +34,15 @@ class Generator(nn.Module):
         # Initial convolution block
         model0 = [
             nn.ReflectionPad2d(3),
-            nn.Conv2d(input_nc, 256, 7),
-            norm_layer(256),
+            nn.Conv2d(input_nc, 64, 7),
+            norm_layer(64),
             nn.ReLU(inplace=True),
         ]
         self.model0 = nn.Sequential(*model0)
 
         # Downsampling
         model1 = []
-        in_features = 256
+        in_features = 64
         out_features = in_features * 2
         for _ in range(2):
             model1 += [