Email_Detector / app.py
Rehman1603's picture
Update app.py
3f45e86
raw
history blame contribute delete
803 Bytes
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)