Skip to content

Commit 7a15db8

Browse files
committed
Load default segmentation model correctly when none is given
Fixes #751
1 parent 51f84c9 commit 7a15db8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

kraken/blla.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ def segment(im: PIL.Image.Image,
299299
"""
300300
if model is None:
301301
logger.info('No segmentation model given. Loading default model.')
302-
model = vgsl.TorchVGSLModel.load_model(resources.files(__name__).joinpath('blla.mlmodel'))
302+
model = vgsl.TorchVGSLModel.load_model(resources.files('kraken').joinpath('blla.mlmodel'))
303303

304304
if isinstance(model, vgsl.TorchVGSLModel):
305305
model = [model]

0 commit comments

Comments
 (0)