Spaces:
Sleeping
Sleeping
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) |