shubhamjaiswar commited on
Commit
b738a54
·
1 Parent(s): 500b18c

Upload 6 files

Browse files
Files changed (6) hide show
  1. app.py +25 -0
  2. p_non_spam.pkl +3 -0
  3. p_spam.pkl +3 -0
  4. parameters_non_spam.pkl +3 -0
  5. parameters_spam.pkl +3 -0
  6. 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