import streamlit as st
import toxic
import gpt2

PAGES = {
    "Toxic Comment Classifier": toxic,
    "GPT-2 Text Generator": gpt2
}

st.sidebar.title('Navigation')
selection = st.sidebar.radio("Go to", list(PAGES.keys()))

page = PAGES[selection]

page.app()