|
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() |