INNER CODE UNIT · Python
load_model
interpretml/interpret-text · python/interpret_text/generative/model_lib/app.py:16
def load_model(model_name):
return get_model(model_name)
# Sidebar
st.sidebar.header('User Input Parameters')
# Model selection
selected_model = st.sidebar.selectbox(
'Select a model', list(MODEL_LIST.keys()))
model = load_model(selected_model)
datasets = [f for f in os.listdir(dataset_folder) if f.endswith('.csv')]
selected_dataset = st.sidebar.selectbox('Select a dataset', datasets)
# Load the selected dataset
data = pd.read_csv(f'{dataset_folder}/{selected_dataset}', header=0)