From ef994e09872bcc3b4b34bd63d1e8302fff2bc0b2 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 12 Jun 2018 17:12:17 +0200 Subject: spaces... wtf --- models/models.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'models/models.py') diff --git a/models/models.py b/models/models.py index 8e72e46..f5bc8d0 100755 --- a/models/models.py +++ b/models/models.py @@ -4,14 +4,14 @@ import torch def create_model(opt): if opt.model == 'pix2pixHD': - from .pix2pixHD_model import Pix2PixHDModel, InferenceModel - if opt.isTrain: + from .pix2pixHD_model import Pix2PixHDModel, InferenceModel + if opt.isTrain: model = Pix2PixHDModel() else: model = InferenceModel() else: - from .ui_model import UIModel - model = UIModel() + from .ui_model import UIModel + model = UIModel() model.initialize(opt) if opt.verbose: print("model [%s] was created" % (model.name())) -- cgit v1.2.3-70-g09d2