diff --git a/app/apps/core/tasks.py b/app/apps/core/tasks.py
index bc7a008ad54ce1c5b6dd67d0c373b1d54bf16f9e..0480f7de027f7f6f0a3d884cf902b68ccaf8a7f9 100644
--- a/app/apps/core/tasks.py
+++ b/app/apps/core/tasks.py
@@ -146,8 +146,7 @@ def segtrain(task, model_pk, document_pk, part_pks, user_pk=None):
         load = model.file.path
         upload_to = model.file.field.upload_to(model, model.name + '.mlmodel')
     except ValueError:  # model is empty
-        # load = settings.KRAKEN_DEFAULT_SEGMENTATION_MODEL
-        load = None
+        load = settings.KRAKEN_DEFAULT_SEGMENTATION_MODEL
         upload_to = model.file.field.upload_to(model, model.name + '.mlmodel')
         model.file = upload_to
 
@@ -185,6 +184,7 @@ def segtrain(task, model_pk, document_pk, part_pks, user_pk=None):
             evaluation_data=evaluation_data,
             threads=LOAD_THREADS,
             augment=True,
+            resize='both',
             load_hyper_parameters=True)
 
         if not os.path.exists(os.path.split(modelpath)[0]):