Update app.py

app.py

@@ -36,14 +36,17 @@ st.title('Prediksi Sentimen menggunakan BERT')
 text = st.text_area('Masukkan teks', '')
 
 # Tombol untuk memprediksi sentimen
-if st.button('Cek'):
+if st.button('Prediksi'):
     if text.strip() == '':
         st.warning('Masukkan teks terlebih dahulu.')
-        st.write
     else:
-        sentiment = predict_sentiment(text)
-        st.write("Hasil Sentiment :")
+        sentiment, probabilities = predict_sentiment(text)
+        
+        # Menghitung persentase probabilitas sentimen positif
+        positive_probability = probabilities[1] * 100
+        negative_probability = probabilities[0] * 100
+        st.write(f'HASIL PREDIKSI')
         if sentiment == 0:
-            st.write('Negatif')
+            st.write(f'Negatif ({negative_probability:.2f}%)')
         else:
-            st.write('Positif')
+            st.write(f'Positif ({positive_probability:.2f}%)')