NLP_app / app.py
Awlly's picture
minor adjustments
48ac93e
raw
history blame
523 Bytes
import streamlit as st
from transformers import AutoTokenizer, AutoModelForSequenceClassification
from app_pages import page1_model_comparison, page2_rubert_toxicity, page3_gpt_model
st.sidebar.title('Navigation')
selection = st.sidebar.radio("Go to", ["Model Comparison", "RuBERT Toxicity Detection", "GPT Model"])
if selection == "Model Comparison":
page1_model_comparison.run()
elif selection == "RuBERT Toxicity Detection":
page2_rubert_toxicity.run()
elif selection == "GPT Model":
page3_gpt_model.run()