lee45 commited on
Commit
cdea21f
·
1 Parent(s): f714d50

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import joblib
3
+ load_model,load_vectorizer = joblib.load('spam_detection.pkl')
4
+ def classify(text):
5
+ temp = [text]
6
+ temp_data_F = load_vectorizer.transform(temp)
7
+ t = load_model.predict(temp_data_F)[0]
8
+ if(t == 0):
9
+ return("Not A SPAM")
10
+ else:
11
+ return("SPAM")
12
+ gr.Interface(fn=classify,inputs='text',outputs="text",title = "SPAM EMAIL DETECTOR").launch()