Update visualize.py

src/visualization/visualize.py

@@ -1 +1,2 @@
 import streamlit as st
+import yaml