TURNA / app.py
yirmibesogluz's picture
Added classification tasks
9ab8c93
raw
history blame
1.07 kB
import awesome_streamlit as ast
import streamlit as st
from transformers import pipeline
import apps.summarization
import apps.home
import apps.paraphrasing
import apps.title_generation
import apps.sentiment
import apps.categorization
st.set_page_config(
page_title="Turna",
page_icon="πŸ“–",
layout='wide'
)
PAGES = {
"Turna": apps.home,
"Text Summarization": apps.summarization,
"Text Paraphrasing": apps.paraphrasing,
"News Title Generation": apps.title_generation,
"Sentiment Classification": apps.sentiment,
"Text Categorization": apps.categorization,
}
st.sidebar.title("Navigation")
selection = st.sidebar.radio("Pages", list(PAGES.keys()))
page = PAGES[selection]
# with st.spinner(f"Loading {selection} ..."):
ast.shared.components.write_page(page)
st.sidebar.header("Info")
st.sidebar.write(
"Models are available on [HF Hub](https://huggingface.co/collections/boun-tabi-LMG)"
)
st.sidebar.write(
"Model source code available on [GitHub](https://github.com/boun-tabi-LMG/turkish-lm-tuner)"
)