diff --git a/demo.ipynb b/demo.ipynb index ef1a231..cb565dc 100644 --- a/demo.ipynb +++ b/demo.ipynb @@ -42,7 +42,7 @@ "outputs": [], "source": [ "# load config\n", - "cfg = yaml.load(open('configs/mb1_120x120.yml'), Loader=yaml.FullLoader)\n", + "cfg = yaml.load(open('configs/mb1_120x120.yml'), Loader=yaml.SafeLoader)\n", "tddfa = TDDFA(gpu_mode=False, **cfg)\n", "\n", "# Initialize FaceBoxes\n",