Jayesh13 commited on
Commit
f7bdf42
1 Parent(s): 491a475

Upload 2 files

Browse files
Files changed (2) hide show
  1. spam.py +33 -0
  2. spam_model.sav +0 -0
spam.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pickle
3
+ import numpy as np
4
+ import pandas as pd
5
+ import nltk
6
+ import re
7
+ from nltk.corpus import stopwords
8
+ from nltk.stem.porter import PorterStemmer
9
+ from nltk.stem import WordNetLemmatizer
10
+ import sklearn
11
+ from sklearn import feature_extraction
12
+ from sklearn.feature_extraction.text import CountVectorizer
13
+ from sklearn.feature_extraction.text import TfidfVectorizer
14
+
15
+ model = pickle.load(open('C:\Users\HP\Documents\Model_deployment\spam_model.sav','rb'))
16
+
17
+ st.title('Spam classifier')
18
+ input = st.text_area('Enter sms here')
19
+ ps = PorterStemmer()
20
+ review = re.sub('[^a-zA-Z]', ' ', input) # to remove punctuation
21
+ review = review.lower()
22
+ review = review.split()
23
+ review = [ps.stem(word) for word in review if not word in stopwords.words('english')]
24
+ review = ' '.join(review)
25
+
26
+ tf = TfidfVectorizer()
27
+ X = tf.fit_transform(review).toarray()
28
+
29
+
30
+
31
+ if input:
32
+ output = model.predict(X)
33
+ st.json(output)
spam_model.sav ADDED
Binary file (50.6 kB). View file