Commit
·
b738a54
1
Parent(s):
500b18c
Upload 6 files
Browse files- app.py +25 -0
- p_non_spam.pkl +3 -0
- p_spam.pkl +3 -0
- parameters_non_spam.pkl +3 -0
- parameters_spam.pkl +3 -0
- requirements.txt +4 -0
app.py
ADDED
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
import re
|
3 |
+
import joblib
|
4 |
+
p_spam = joblib.load("p_spam.pkl")
|
5 |
+
p_non_spam = joblib.load("p_non_spam.pkl")
|
6 |
+
parameters_spam = joblib.load("parameters_spam.pkl")
|
7 |
+
parameters_non_spam = joblib.load("parameters_non_spam.pkl")
|
8 |
+
def classify(message):
|
9 |
+
message = re.sub("\W", " ", message)
|
10 |
+
message = message.lower().split()
|
11 |
+
p_spam_given_message = p_spam
|
12 |
+
p_non_spam_given_message = p_non_spam
|
13 |
+
for word in message:
|
14 |
+
if word in parameters_spam:
|
15 |
+
p_spam_given_message *= parameters_spam[word]
|
16 |
+
if word in parameters_non_spam:
|
17 |
+
p_non_spam_given_message *= parameters_non_spam[word]
|
18 |
+
if p_spam_given_message > p_non_spam_given_message:
|
19 |
+
return "Spam"
|
20 |
+
elif p_non_spam_given_message > p_spam_given_message:
|
21 |
+
return "Non-Spam"
|
22 |
+
else:
|
23 |
+
return "Equal probabilities, human needed to classify this!"
|
24 |
+
ir = gr.Interface(classify,inputs="text",outputs="text")
|
25 |
+
ir.launch()
|
p_non_spam.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd63c725e9626560f36b327a0f3864a8392d5d7665a69f81bbae95231242c4d9
|
3 |
+
size 21
|
p_spam.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1b44f97c87b342e7764d3da770f79569c2a93d0663af741f297ed873fb983e3a
|
3 |
+
size 21
|
parameters_non_spam.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5205c780118f80d13daed2516c16db9fca1e905a6064a00c110e4cdf2cb67a3b
|
3 |
+
size 216055
|
parameters_spam.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:69368b482721c833a9ec39077fd34b8c425c81163e26103fceb438c24e0b451a
|
3 |
+
size 216055
|
requirements.txt
ADDED
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
1 |
+
re
|
2 |
+
joblib
|
3 |
+
gradio
|
4 |
+
pandas
|