notebook kernel dies

[fasahath]

Notebook kernel dies when calling model.detect_image_show(image_path)

GPU Details:
NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2

Torch Version: 1.5.0
facemask Version: 0.1