nlp-lstm-team / app.py
Rzhishchev's picture
Create app.py
37669fc
raw
history blame
No virus
259 Bytes
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()