From f0156312e414a1b4a5c1078e000c1c50b9243d8e Mon Sep 17 00:00:00 2001
From: Ashish Sinha <ashishsinha108@gmail.com>
Date: Mon, 20 Feb 2023 22:48:03 -0800
Subject: [PATCH] fixes yaml loader bug

---
 utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index a0e4349..6e269d6 100644
--- a/utils.py
+++ b/utils.py
@@ -12,7 +12,7 @@
 
 def get_config(config):
     with open(config, 'r') as stream:
-        return yaml.load(stream)
+        return yaml.safe_load(stream)
 
 def prepare_sub_folder(output_directory):
     image_directory = os.path.join(output_directory, 'images')