Update app.py

app.py

@@ -1,18 +1,10 @@
-"""
-#* make sure to run the application using this commmand:
->>> streamlit run main.py
-"""
-
-# global
 import streamlit as st
 from lime.lime_text import LimeTextExplainer
 from nltk.corpus import stopwords
 
-# local
 from deployment_utils import DataPreparator, Predictor, generate_random_sample, generate_highlighted_words, extract_case_information
 
 
-# instantiate `DataPreparator` & `Predictor` objects
 data_preparator = DataPreparator()
 predictor = Predictor()
 eng_stop_words = stopwords.words("english")
@@ -22,7 +14,6 @@ st.set_page_config(
     page_icon="🧊",
     layout="wide")
 
-# for custom CSS styling
 with open("./src/style.css") as f:
     st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
 
@@ -106,18 +97,9 @@ with right_col:
 
         option = st.selectbox(
             "Select Model",
-            ("TF-IDF", "1D Convolutional", "GloVe", "BERT", "Doc2Vec",
-             "LSTM", "FastText", "Ensemble (Doc2Vec + TF-IDF)")
+            ("CNN","BERT","Doc2Vec")
         )
 
-        # if `Get Random Sample` btn is pressed
-        # if get_random_case_button:
-        #     random_petitioner, random_respondent, random_facts, random_label = generate_random_sample()
-        #     st.session_state["petitioner"] = random_petitioner.strip()
-        #     st.session_state["respondent"] = random_respondent.strip()
-        #     st.session_state["facts"] = random_facts.strip()
-        #     st.success(f"Original label: {random_label}")
-
         col1, col2 = st.columns(2)
 
         with col1:
@@ -248,7 +230,7 @@ with right_col:
 
                 else:
                     st.warning(
-                        "Sadly, this feature is not supported in BERT & FastText :(")
+                        "Sadly, this feature is not supported in BERT")
 
             else:
                 st.error("Please, fill in all fields!")