to 512

app.py

@@ -76,7 +76,7 @@ class Generator(nn.Module):
         self.model3 = nn.Sequential(*model3)
 
         # Output layer
-        model4 = [nn.ReflectionPad2d(3), nn.Conv2d(256, output_nc, 7)]
+        model4 = [nn.ReflectionPad2d(3), nn.Conv2d(64, output_nc, 7)]
         if sigmoid:
             model4 += [nn.Sigmoid()]
 
@@ -104,7 +104,7 @@ model2.eval()
 def predict(input_img, ver):
     input_img = Image.open(input_img)
     transform = transforms.Compose(
-        [transforms.Resize(1080, Image.BICUBIC), transforms.ToTensor()]
+        [transforms.Resize(512, Image.BICUBIC), transforms.ToTensor()]
     )
     input_img = transform(input_img)
     input_img = torch.unsqueeze(input_img, 0)