File size: 803 Bytes
3bb52d2
 
 
 
3f45e86
 
 
98b0652
3bb52d2
44fb3a5
3bb52d2
 
 
 
 
3f45e86
 
d44f066
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import pickle
import gradio as gr
from sklearn.feature_extraction.text import TfidfVectorizer

with open("Mail_Pickle","rb") as f:
  mp=pickle.load(f)

feature_extraction=TfidfVectorizer(min_df=1,stop_words='english',lowercase=True)
def Mail_Prediction(Email):
  input_Email_conv=feature_extraction.fit_transform([Email])
  result=mp.predict(input_Email_conv)
  if(result[0]==1):
    final_result="HAM"
    return final_result
  elif(result[0]==0):
    final_result="SPAM"
    return final_result


interface=gr.Interface(fn=Mail_Prediction,inputs=[gr.inputs.Textbox(lines=2,placeholder="Enter your Email",label="Spam Email Checker")],
                       outputs=[gr.outputs.Textbox(label="Your result")],
                       enable_queu=True
                       )
interface.launch(debug=True)