Update app.py
Browse files
app.py
CHANGED
@@ -1,34 +1,34 @@
|
|
1 |
-
# import libraries
|
2 |
-
import streamlit as st
|
3 |
-
import predict
|
4 |
-
import logging
|
5 |
-
|
6 |
-
# Logging configuration
|
7 |
-
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
|
8 |
-
|
9 |
-
# Load necessary components
|
10 |
-
df = predict.load_dataset()
|
11 |
-
model, tokenizer = predict.load_model_and_tokenizer()
|
12 |
-
|
13 |
-
if df is not None and model is not None and tokenizer is not None:
|
14 |
-
# Preprocess dataset and prepare stopwords and stemmer
|
15 |
-
manual_stopwords = ["di", "ke", "dari", "yang", "dan", "atau", "dengan", "untuk", "ini", "itu", "aja", "saja", "lah", "bri", "brimo", "aplikasi", "rekening", "coba", "yg", "ke", "untuk", "nya", "saya", "dia", "dan", "sangat", "video", "login", "apk", "jadi", "akun", "malah", "uang", "banget", "dalam", "atm", "padahal"]
|
16 |
-
stop_words = set(predict.stopwords.words('indonesian'))
|
17 |
-
stop_words.update(manual_stopwords)
|
18 |
-
factory = predict.StemmerFactory()
|
19 |
-
stemmer = factory.create_stemmer()
|
20 |
-
df['label'] = df['score'].apply(predict.map_labels)
|
21 |
-
|
22 |
-
navigation = st.sidebar.selectbox('Choose Page:', ('predictor'
|
23 |
-
|
24 |
-
if navigation == 'predictor':
|
25 |
-
predict.run(model, tokenizer, stop_words, stemmer)
|
26 |
-
# elif navigation == 'eda':
|
27 |
-
# eda.run()
|
28 |
-
else:
|
29 |
-
if df is None:
|
30 |
-
logging.error("Failed to load dataset.")
|
31 |
-
st.error("Failed to load the dataset. Please check the dataset file.")
|
32 |
-
if model is None or tokenizer is None:
|
33 |
-
logging.error("Failed to load model or tokenizer.")
|
34 |
st.error("Failed to load the model or tokenizer. Please check the model file.")
|
|
|
1 |
+
# import libraries
|
2 |
+
import streamlit as st
|
3 |
+
import predict
|
4 |
+
import logging
|
5 |
+
|
6 |
+
# Logging configuration
|
7 |
+
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
|
8 |
+
|
9 |
+
# Load necessary components
|
10 |
+
df = predict.load_dataset()
|
11 |
+
model, tokenizer = predict.load_model_and_tokenizer()
|
12 |
+
|
13 |
+
if df is not None and model is not None and tokenizer is not None:
|
14 |
+
# Preprocess dataset and prepare stopwords and stemmer
|
15 |
+
manual_stopwords = ["di", "ke", "dari", "yang", "dan", "atau", "dengan", "untuk", "ini", "itu", "aja", "saja", "lah", "bri", "brimo", "aplikasi", "rekening", "coba", "yg", "ke", "untuk", "nya", "saya", "dia", "dan", "sangat", "video", "login", "apk", "jadi", "akun", "malah", "uang", "banget", "dalam", "atm", "padahal"]
|
16 |
+
stop_words = set(predict.stopwords.words('indonesian'))
|
17 |
+
stop_words.update(manual_stopwords)
|
18 |
+
factory = predict.StemmerFactory()
|
19 |
+
stemmer = factory.create_stemmer()
|
20 |
+
df['label'] = df['score'].apply(predict.map_labels)
|
21 |
+
|
22 |
+
navigation = st.sidebar.selectbox('Choose Page:', ('predictor'))
|
23 |
+
|
24 |
+
if navigation == 'predictor':
|
25 |
+
predict.run(model, tokenizer, stop_words, stemmer)
|
26 |
+
# elif navigation == 'eda':
|
27 |
+
# eda.run()
|
28 |
+
else:
|
29 |
+
if df is None:
|
30 |
+
logging.error("Failed to load dataset.")
|
31 |
+
st.error("Failed to load the dataset. Please check the dataset file.")
|
32 |
+
if model is None or tokenizer is None:
|
33 |
+
logging.error("Failed to load model or tokenizer.")
|
34 |
st.error("Failed to load the model or tokenizer. Please check the model file.")
|