Jayesh13 commited on
Commit
b8e812e
1 Parent(s): 1307ab5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,6 +11,7 @@ from sklearn.feature_extraction.text import CountVectorizer
11
  from sklearn.feature_extraction.text import TfidfVectorizer
12
 
13
  model = pickle.load(open('spam_model.sav','rb'))
 
14
 
15
  st.title('Spam classifier')
16
  input = st.text_area('Enter sms here')
@@ -21,7 +22,7 @@ review = review.split()
21
  review = [ps.stem(word) for word in review if not word in stopwords.words('english')]
22
  review = ' '.join(review)
23
 
24
- tf = TfidfVectorizer()
25
  X = tf.fit_transform(review).toarray()
26
 
27
 
 
11
  from sklearn.feature_extraction.text import TfidfVectorizer
12
 
13
  model = pickle.load(open('spam_model.sav','rb'))
14
+ tf = pickle.load(open('vectorizer.pkl','rb'))
15
 
16
  st.title('Spam classifier')
17
  input = st.text_area('Enter sms here')
 
22
  review = [ps.stem(word) for word in review if not word in stopwords.words('english')]
23
  review = ' '.join(review)
24
 
25
+
26
  X = tf.fit_transform(review).toarray()
27
 
28