use yaml.safe_load()

TypeError: load() missing 1 required positional argument: 'Loader'
YuanxunLu · May 16, 2022 · a2f24cf · a2f24cf
1 parent c848865
commit a2f24cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -48,7 +48,7 @@ def predict(self,
         opt.driving_audio = str(driving_audio)
         opt.id = talking_head
         with open(join('config', opt.id + '.yaml')) as f:
-            config = yaml.load(f)
+            config = yaml.safe_load(f)
         data_root = join('data', opt.id)
 
         ############################ Hyper Parameters #############################