Review_Verifier / app.py
muruga778's picture
Create app.py
130dc51 verified
raw
history blame contribute delete
729 Bytes
import gradio as gr
from joblib import load
# Load the vectorizer and model
vectorizer = load('tfidf_vectorizer.pkl')
model = load('fake_review_detector.pkl')
# Prediction function
def predict_review(review):
# Transform the input text into TF-IDF
review_tfidf = vectorizer.transform([review])
# Make a prediction
prediction = model.predict(review_tfidf)[0]
# Return the result
return "THIS IS NOT FAKE" if prediction == 0 else "THIS IS FAKE"
# Gradio interface
interface = gr.Interface(
fn=predict_review,
inputs="text",
outputs="text",
title="Fake Review Detector",
description="Type a review below to determine if it is fake or not."
)
# Launch the interface
interface.launch()