diff --git a/neural_style_app/mode_style_transfer.py b/neural_style_app/mode_style_transfer.py index fadfecf..b8c0705 100644 --- a/neural_style_app/mode_style_transfer.py +++ b/neural_style_app/mode_style_transfer.py @@ -157,7 +157,6 @@ class StyleTransferModel: def __init__(self, content_img, style_img, num_steps=300, style_weight=1000000, content_weight=1): self.content_img = content_img self.style_img = style_img.resize(content_img.size) - #self.style_img = self.style_img.resize(self.content_img.size) self.style_img = image_loader(self.style_img) self.content_img = image_loader(self.content_img) self.input_img = self.content_img.clone()