theArijitDas's picture
Upload 2 files
7d596e1 verified
from transformers import pipeline
import gradio as gr
model = pipeline("text-classification", model="theArijitDas/distilbert-finetuned-fake-reviews-dataset")
# List of AI-related keywords
ai_keywords = [
"AI chatbot", "Generated by AI", "Machine learning", "Artificial intelligence", "Deep learning",
"As a language model", "Powered by AI", "As an AI assistant", "Trained by neural networks",
"As an AI model", "Based on AI technology", "AI-generated", "AI-powered", "Algorithmic content",
"Automated text", "Computer-generated", "Generated by machine", "Created by AI", "Synthetic text",
"AI-driven", "Using natural language processing", "As an AI language model", "AI system", "AI-based",
"AI assistance", "Generated by machine learning", "AI capabilities", "AI algorithms", "Produced by AI",
"Automated response", "As an artificial intelligence", "With AI support", "AI-generated content",
"AI-enhanced", "Using deep learning", "Generated using algorithms", "Created by machine learning",
"Produced by machine learning", "Automated generation", "Computer-generated response", "AI-based content",
"AI-generated text", "Machine-produced", "AI-processed", "AI-generated response", "With machine learning",
"By artificial intelligence", "AI-created", "Neural network", "Generated by a computer",
"Automated content creation", "Created by algorithms", "Synthesized text", "AI-crafted",
"Text produced by AI", "AI content generation", "Programmatically generated", "Generated by neural networks",
"AI-assisted", "AI text generator", "Generated by NLP", "As a neural network", "AI text creation",
"Machine learning model", "AI-generated language", "Automated writing", "NLP-generated", "Text by AI",
"AI-produced", "AI-enabled", "Algorithm-driven content", "As an intelligent system", "Machine-authored",
"AI text", "AI machine learning", "AI language processing", "Text generated by AI", "Automated AI",
"AI language model", "Generated with AI", "Algorithmic text", "AI-authored", "With artificial intelligence",
"By machine learning algorithms", "AI-generated phrases", "Computer-created", "AI neural networks",
"Machine-generated text", "Automated neural network", "AI content", "AI-powered generation",
"AI textual content", "AI natural language", "Machine-generated content", "AI-driven text",
"Neural network-generated", "AI output", "Generated by artificial intelligence", "Automated generation by AI",
"AI synthetic text", "AI bot",
]
def contains_ai_keywords(text):
for keyword in ai_keywords:
if keyword.lower() in text.lower():
return True
return False
def classify_review_with_model(text):
label = model(text)[0]['label']
return label
def classify(text):
if contains_ai_keywords(text):
return "FAKE" # Flag as AI-generated
else:
return classify_review_with_model(text)
iface = gr.Interface(classify,
inputs=["text"],
outputs=["text"],
title="Fake Review Detector",
description="Enter product review to classify it as REAL or FAKE.")
iface.launch()