Spaces:
Sleeping
Sleeping
1st commit
Browse files- SMS_Classification.ipynb +0 -0
- app.py +23 -0
- model.pkl +3 -0
- spam_sms.csv +0 -0
- vectorizer.pkl +3 -0
SMS_Classification.ipynb
ADDED
The diff for this file is too large to render.
See raw diff
|
|
app.py
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import pickel
|
3 |
+
|
4 |
+
#lets load the saved vectorizer and native model
|
5 |
+
tfidf = pickel.load(open('vectorizer.pkl','rb'))
|
6 |
+
model = pickel.load(open('model.pkl','rb'))
|
7 |
+
|
8 |
+
#saving streamlit code
|
9 |
+
st.title("Email spam Classifier")
|
10 |
+
input_sms = st.text_area("Enter message")
|
11 |
+
|
12 |
+
if st.button('predict'):
|
13 |
+
#preprocess
|
14 |
+
transformed_sms = transform_text(input_sms)
|
15 |
+
#vectorize
|
16 |
+
vector_input = tfidf.transform([transformed_sms])
|
17 |
+
#predict
|
18 |
+
result = model.predict(vector_input)[0]
|
19 |
+
#display
|
20 |
+
if result ==1:
|
21 |
+
st.header("Spam")
|
22 |
+
else:
|
23 |
+
st.header("Not Spam")
|
model.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d353a616171e314953eabebc9a78df13bb413ce897405b9a2f75bf66628f6b88
|
3 |
+
size 96613
|
spam_sms.csv
ADDED
The diff for this file is too large to render.
See raw diff
|
|
vectorizer.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d3d66f70e58fa10dea2bea3fc692e190473de6c8471bc8d1973b0f9f70046f7
|
3 |
+
size 181749
|