src/visualization/visualize.py

import streamlit as st
import yaml

from models import predict_model