nab3588 commited on
Commit
5486bbd
·
verified ·
1 Parent(s): 4873685

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -6
src/streamlit_app.py CHANGED
@@ -22,13 +22,13 @@ st.set_page_config(page_title="Warung Sentiment Dashboard", page_icon="🍜", la
22
  st.title("Analisis Kepuasan Pelanggan")
23
  st.markdown("Masukkan hingga **15 ulasan pelanggan** untuk mengetahui tingkat kepuasan mereka secara mudah dan visual.")
24
 
25
- # --- LOAD MODEL (CACHED) ---
26
  @st.cache_resource
27
  def load_model():
28
- model_name = "w11wo/indonesian-roberta-base-sentiment-classifier"
29
- tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
30
- model = AutoModelForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
31
- return pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
32
 
33
  sentiment_model = load_model()
34
 
@@ -178,7 +178,7 @@ if submitted and len(comments) > 0:
178
  "makannya", "masakannya", "minumannya",
179
 
180
  # Redundant sentiment intensifiers
181
- "sekali", "banget", "parah", "lumayan", "agak", "terlalu", "cukup", "amat", "sih",
182
 
183
  # Common verbs that don't carry meaning in context
184
  "adalah", "merupakan", "dapat", "bikin", "buat", "kasih", "dikasih", "punya", "bisa",
 
22
  st.title("Analisis Kepuasan Pelanggan")
23
  st.markdown("Masukkan hingga **15 ulasan pelanggan** untuk mengetahui tingkat kepuasan mereka secara mudah dan visual.")
24
 
 
25
  @st.cache_resource
26
  def load_model():
27
+ from transformers import pipeline
28
+ return pipeline(
29
+ "sentiment-analysis",
30
+ model="w11wo/indonesian-roberta-base-sentiment-classifier",
31
+ )
32
 
33
  sentiment_model = load_model()
34
 
 
178
  "makannya", "masakannya", "minumannya",
179
 
180
  # Redundant sentiment intensifiers
181
+ "sekali", "banget", "parah", "lumayan", "agak", "terlalu", "cukup", "amat", "sih", "tetapi"
182
 
183
  # Common verbs that don't carry meaning in context
184
  "adalah", "merupakan", "dapat", "bikin", "buat", "kasih", "dikasih", "punya", "bisa",