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