TURNA / app.py
gokceuludogan's picture
Update app.py
fe3e95d verified
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
import apps.ner
import apps.pos_tagging
import apps.sts
import apps.nli
import apps.turna
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,
"Named Entity Recognition": apps.ner,
"Part-of-Speech Tagging": apps.pos_tagging,
"Semantic Textual Similarity": apps.sts,
"Natural Language Inference": apps.nli,
"Text Generation": apps.turna,
}
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)"
)