Beasto commited on
Commit
06f4e51
·
1 Parent(s): 99fda08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,10 +3,10 @@ from sklearn.feature_extraction.text import TfidfVectorizer
3
  import numpy as np
4
  import streamlit as st
5
 
6
- vectorizer = TfidfVectorizer(use_idf=True, lowercase=True, strip_accents='ascii', stop_words='english')
7
- user_input = st.text_input("Enter some text:", "Default Text")
8
  if user_input is not None:
9
- x = vectorizer.fit_transform(np.array(user_input))
10
  model = load('Naive_Bayes_Spam_Detection.joblib')
11
  pred = model.predict(x)
12
  st.write(pred)
 
3
  import numpy as np
4
  import streamlit as st
5
 
6
+ vectorizer = load('tfidf_vectorizer.joblib')
7
+ user_input = st.text_input("Enter some text:", "")
8
  if user_input is not None:
9
+ x = vectorizer.transform(np.array(user_input))
10
  model = load('Naive_Bayes_Spam_Detection.joblib')
11
  pred = model.predict(x)
12
  st.write(pred)