Test / Model.py
Riyan00b's picture
Upload Model.py
4dd8f0b verified
#!/usr/bin/env python
# coding: utf-8
# In[10]:
import gradio as gr
from joblib import load
model = load('model.joblib')
conv = load('tfd.joblib')
def prediction(email):
inp = [email]
inp_final = conv.transform(inp)
res = model.predict(inp_final)[0]
return "not spam" if res == 1 else 'spam'
iface = gr.Interface(
fn = prediction,
inputs = [gr.Text(label="Email")],
outputs = 'text',
title = 'spam Identifier',
description = 'spam detection')
iface.launch()
# In[9]:
# In[ ]: