Update visualize.py

src/visualization/visualize.py

@@ -1,3 +1,4 @@
 import streamlit as st
 import yaml
 
+from models import predict_model