SpamChecker / app.py
drmurataltun's picture
Create app.py
99dec5d verified
raw
history blame contribute delete
694 Bytes
import gradio as gr
import pickle
# metin sınıflandırıcıyı ve vektörleştiriciyi ayrı ayrı kaydet
model=pickle.load(open('bnb_tc.pkl','rb'))
vect=pickle.load(open('vect.pkl','rb'))
def predict_spam(review):
#yorumu vektöre dönüştür
vect_text=vect.transform([review])
#tahmin yap
prediction=model.predict(vect_text)
return prediction[0]
# Ara yüz oluştur
interface=gr.Interface(
fn=predict_spam,
inputs=gr.Textbox(lines=2,
placeholder="Enter a review..."),
outputs='text',
title="YouTube Comment Spam Detector",
description="This tool detect whether YouTube comment a spam."
)
interface.launch(
share=True
)