Update app.py

app.py

@@ -19,7 +19,7 @@ pad=np.pad
 def load_checkpoints(config_path, checkpoint_path, cpu=False):
 
     with open(config_path) as f:
-        config = yaml.load(f)
+        config = yaml.load(f, Loader=yaml.SafeLoader)
 
     generator = OcclusionAwareGenerator(
         **config["model_params"]["generator_params"], **config["model_params"]["common_params"]