from ArtistCoherencyModel import ArtistCoherencyModel
import streamlit as st
import pandas as pd

artists_df = pd.read_csv("artists.csv")
artist_names_list = list(artists_df["name"])


artist_name_input = st.selectbox("Artist", artist_names_list)
st.write(artist_name_input)

ensemble_model = ArtistCoherencyModel.from_pretrained(
    "tjl223/artist-coherency-ensemble"
)