Update app.py

app.py

@@ -25,12 +25,16 @@ if st.button("ตรวจสอบ"):
         
         # แสดงค่าความมั่นใจ
         confidence = result[0]['score'] * 100  # แปลงค่าความมั่นใจเป็นเปอร์เซ็นต์
-        st.write(f"แสปม: {result[0]['label']} ({confidence:.2f}% ความมั่นใจ)")
+        label = result[0]['label']
 
-        # สร้างเงื่อนไขในการตัดสินใจ
-        if confidence >= 70:
-            st.warning("โมเดลมั่นใจว่าข้อความนี้เป็นสแปม")
+        if label == "LABEL_FOR_SPAM":  # แทน LABEL_FOR_SPAM ด้วยป้ายกำกับสแปมของโมเดลของคุณ
+            st.write(f"สแปม: {label} ({confidence:.2f}% ความมั่นใจ)")
+            if confidence >= 70:
+                st.warning("โมเดลมั่นใจว่าข้อความนี้เป็นสแปม")
+            else:
+                st.success("โมเดลมั่นใจว่าข้อความนี้ไม่เป็นสแปม")
         else:
+            st.write(f"ไม่เป็นสแปม: {label} ({confidence:.2f}% ความมั่นใจ)")
             st.success("โมเดลมั่นใจว่าข้อความนี้ไม่เป็นสแปม")
     else:
         st.warning("โปรดป้อนข้อความก่อนตรวจสอบ")