Rehman1603 commited on
Commit
98b0652
·
1 Parent(s): eb0e7ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,8 +2,9 @@ import pickle
2
  import gradio as gr
3
  from sklearn.feature_extraction.text import TfidfVectorizer
4
 
 
5
  def Mail_Prediction(Email):
6
- input_Email_conv=vectorizer.transform(Email)
7
  result=mp.predict(input_Email_conv)
8
  if(result[0]==1):
9
  final_result="HAM"
 
2
  import gradio as gr
3
  from sklearn.feature_extraction.text import TfidfVectorizer
4
 
5
+ feature_extraction=TfidfVectorizer(min_df=1,stop_words='english',lowercase=True)
6
  def Mail_Prediction(Email):
7
+ input_Email_conv=feature_extraction.transform([Email])
8
  result=mp.predict(input_Email_conv)
9
  if(result[0]==1):
10
  final_result="HAM"