Update app.py

app.py

@@ -16,7 +16,7 @@ stemmer = StemmerFactory().create_stemmer()  # Membuat stemmer Sastrawi
 
 def preprocess_text(text):
     # Menggunakan Sastrawi untuk stemming
-    stemmed_text = stemmer.stem(text)
+    stemmed_text = stemmer.stem(text.lower())
 
     return stemmed_text
 
@@ -36,12 +36,14 @@ st.title('Prediksi Sentimen menggunakan BERT')
 text = st.text_area('Masukkan teks', '')
 
 # Tombol untuk memprediksi sentimen
-if st.button('Prediksi'):
+if st.button('Cek'):
     if text.strip() == '':
         st.warning('Masukkan teks terlebih dahulu.')
+        st.write
     else:
         sentiment = predict_sentiment(text)
+        st.write("Hasil Sentiment :")
         if sentiment == 0:
-            st.error('Sentimen: Negatif')
+            st.write('Negatif')
         else:
-            st.success('Sentimen: Positif')
+            st.write('Positif')