|
import streamlit as st |
|
from st_pages import Page, show_pages, add_page_title |
|
from transformers import pipeline |
|
from datasets import load_dataset |
|
|
|
|
|
class TextClassifier: |
|
bertPipe = pipeline( model="nlp-pw/test-model-3") |
|
|
|
|
|
class AnalysisData: |
|
ds = load_dataset("nlp-pw/Disaster-Tweets-Normalized", revision="main", split='train') |
|
|
|
|
|
def main(): |
|
st.session_state["shared"] = True |
|
st.set_page_config( |
|
page_title="NLP - Project", |
|
page_icon="π€", |
|
) |
|
add_page_title("NLP - Project") |
|
show_pages( |
|
[ |
|
Page("pages/introduction.py", "Intro", "π"), |
|
Page("pages/data_analysis.py", "Data and data", "π"), |
|
Page("pages/run_models.py", "Run models", "π€"), |
|
] |
|
) |
|
|
|
if __name__ == "__main__": |
|
main() |
|
|
|
|