nlp-lstm-team / app.py
Rzhishchev's picture
Update app.py
1d81050
raw
history blame
332 Bytes
import streamlit as st
from gpt2 import app as gpt2_app
from toxic import app as toxic_app
# Sidebar for page selection
page = st.sidebar.selectbox("Choose a page", ["GPT-2", "Toxic Comment Detector"])
# Display the selected page
if page == "GPT-2 Generator":
gpt2_app()
elif page == "Toxic Comment Detector":
toxic_app()