Update visualize.py

src/visualization/visualize.py

@@ -26,3 +26,6 @@ def visualize():
 if __name__ == "__main__":
     with open("params.yml") as f:
         params = yaml.safe_load(f)
+
+    if params['visualise']:
+        visualize()