rsakadewa7 commited on
Commit
2d74fd1
·
verified ·
1 Parent(s): 8499e95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -33
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', 'eda'))
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.")