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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import streamlit as st
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)
 
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([user_input])
10
  model = load('Naive_Bayes_Spam_Detection.joblib')
11
  pred = model.predict(x)
12
  st.write(pred)