Orawan commited on
Commit
10a1e30
·
1 Parent(s): 4d8c883

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -40
app.py DELETED
@@ -1,40 +0,0 @@
1
- import streamlit as st
2
- from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
3
-
4
- tokenizer = AutoTokenizer.from_pretrained("Nma/RuleClassify-Textclassify")
5
- model = AutoModelForSequenceClassification.from_pretrained("Nma/RuleClassify-Textclassify")
6
-
7
- # Create a sentiment analysis pipeline with the explicit tokenizer
8
- classifier = pipeline("text-classification", model=model,tokenizer=tokenizer)
9
-
10
- # สร้างหน้าเว็บ Streamlit
11
- st.title("Spam Detection App")
12
-
13
- # สร้างกล่องข้อความให้ผู้ใช้ป้อนข้อความ
14
- user_input = st.text_input("ใส่ข้อความที่ต้องการตรวจสอบ:")
15
-
16
- # ตรวจสอบข้อความเมื่อผู้ใช้กดปุ่ม "ตรวจสอบ"
17
- if st.button("ตรวจสอบ"):
18
- if user_input:
19
- # ใช้โมเดลตรวจสอบข้อความ
20
- result = classifier(user_input)
21
-
22
- # แสดงผลลัพธ์
23
- st.write("ผลลัพธ์:")
24
- st.write(f"ข้อความ: {user_input}")
25
-
26
- # แสดงค่าความมั่นใจ
27
- confidence = result[0]['score'] * 100 # แปลงค่าความมั่นใจเป็นเปอร์เซ็นต์
28
- label = result[0]['label']
29
-
30
- if label == "LABEL_FOR_SPAM": # แทน LABEL_FOR_SPAM ด้วยป้ายกำกับสแปมของโมเดลของคุณ
31
- st.write(f"สแปม: {label} ({confidence:.2f}% ความมั่นใจ)")
32
- if confidence >= 80:
33
- st.warning("โมเดลมั่นใจว่าข้อความนี้ไม่เป็นสแปม")
34
- else:
35
- st.success("โมเดลมั่นใจว่าข้อความนี้เป็นสแปม")
36
- else:
37
- st.write(f"ไม่เป็นสแปม: {label} ({confidence:.2f}% ความมั่นใจ)")
38
- st.success("โมเดลมั่นใจว่าข้อความนี้ไม่เป็นสแปม")
39
- else:
40
- st.warning("โปรดป้อนข้อความก่อนตรวจสอบ")