import streamlit as st
import yaml

from models import predict_model