|
from transformers import pipeline |
|
import gradio as gr |
|
|
|
model = pipeline("text-classification", model="theArijitDas/distilbert-finetuned-fake-reviews-dataset") |
|
|
|
|
|
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" |
|
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() |