Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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 |
|