geekyrakshit's picture
add: llama guard fine-tuner[WIP]
573a89c
raw
history blame
978 Bytes
import streamlit as st
intro_page = st.Page(
"application_pages/intro_page.py", title="Introduction", icon=":material/guardian:"
)
chat_page = st.Page(
"application_pages/chat_app.py",
title="Playground",
icon=":material/sports_esports:",
)
evaluation_page = st.Page(
"application_pages/evaluation_app.py",
title="Evaluation",
icon=":material/monitoring:",
)
train_classifier_page = st.Page(
"application_pages/train_classifier.py",
title="Train Classifier",
icon=":material/fitness_center:",
)
llama_guard_fine_tuning_page = st.Page(
"application_pages/llama_guard_fine_tuning.py",
title="Fine-Tune LLama Guard",
icon=":material/star:",
)
page_navigation = st.navigation(
[
intro_page,
chat_page,
evaluation_page,
train_classifier_page,
llama_guard_fine_tuning_page,
]
)
st.set_page_config(page_title="Guardrails Genie", page_icon=":material/guardian:")
page_navigation.run()