Ryleeeee commited on
Commit
4da51f1
·
verified ·
1 Parent(s): 4cd5c70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -48,4 +48,7 @@ if st.button("Classify Sentiment"):
48
  # Perform text summarization when the review sentiment is classified as negative
49
  if sentiment_result[1] == 'negative':
50
  category = product_dic[int(product_category(review)["label"].split("_")[1])]
51
- st.write("Category of the faulty product: ", category)
 
 
 
 
48
  # Perform text summarization when the review sentiment is classified as negative
49
  if sentiment_result[1] == 'negative':
50
  category = product_dic[int(product_category(review)["label"].split("_")[1])]
51
+ predic_score = product_category(review)["score"]
52
+ st.write("Category of the faulty product: ", category)
53
+ st.write("Prediction score: ", predic_score)
54
+