|
import gradio as gr |
|
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification |
|
|
|
|
|
|
|
|
|
|
|
|
|
model_name = "distilbert-base-uncased-finetuned-sst-2-english" |
|
tokenizer = AutoTokenizer.from_pretrained(model_name) |
|
model = AutoModelForSequenceClassification.from_pretrained(model_name) |
|
|
|
|
|
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer) |
|
|
|
def classify_text(text): |
|
|
|
result = classifier(text)[0] |
|
label = result['label'] |
|
score = round(result['score'], 4) * 100 |
|
|
|
|
|
if label == "NEGATIVE": |
|
interpretation = "Fictional/Unreliable" |
|
else: |
|
interpretation = "Real/Trustworthy" |
|
|
|
|
|
analysis = f"This article appears to be {interpretation}. " |
|
conjecture = f"I am {score:.2f}% confident in this assessment. " |
|
resolution = "It's always prudent to examine further and cross-check with other reliable sources." |
|
|
|
|
|
human_readable_output = f"{analysis}{conjecture}{resolution}" |
|
return human_readable_output |
|
|
|
|
|
examples = [ |
|
|
|
|
|
["Market Responds Favorably to Central Bank's Move on Interest Rates: In a decisive week for the financial markets, global stocks surged in response to the Central Bank's unexpected move to cut interest rates by 50 basis points. Experts argue that this measure will counter recent deflationary pressures, enhance consumer spending, and ultimately catalyze a much-needed recovery for the year."], |
|
["Major Breakthrough in Renewable Energy with the World's Largest Ocean Turbine: Scientists have achieved a pivotal innovation in wind turbine engineering, inaugurating the world's largest floating wind turbine. Stated to power over 20,000 homes with clean, interminable power, this green initiative underlines a game-changing push towards attaining 2040's solid goal of a low-carbon future."], |
|
|
|
["Dinosaurs, Presumed Extinct, Found Vacationing in Bermuda: A trinity of a newsroom in Bermuda has unsettly brought the world to a gasp, spelling the tale of ten opal-tinted dinosaurs unpersuasively masquerading in guise across the country's remote east. Against all recent accordance, these exploratory frigate's portrayal, namely in their talk - postulated to originate from a covert warren hitherto anonymous to modern study - generated a deep rollick in the pool of beastly sociology and stately native tranquility."], |
|
["Scientists Invent a 'Teleportation Device', Promising Human Commutes from Paris to Tokyo in Seconds: Surging past a pantheon of sci-fi omniscience, today's manuscript from the Multiverse Tinkerer papers spells out a term that witnessed what the previous generations fathomed as providence: the drawing board for a predawn 'Hub-linkage Teleporter'. Affirming to command the focus of figure to the behest of time, this debutante appurtenance avows to break traditions by converging the Eiffel Tower and Tokyo Tower within a nictation's leap, jellifying glances if not the vast potboiler's comprehension of piebald solidity."], |
|
] |
|
|
|
iface = gr.Interface(fn=classify_text, |
|
inputs=gr.components.Textbox(lines=2, placeholder="Input the news article here..."), |
|
outputs="text", |
|
title="Real or Fictional News Recognition", |
|
description="This model is a fine-tuned DistilBERT model for detecting fake news. It was trained on the SST-2 dataset. It distinguishes real news from the fiction. Below are some preloaded examples you can choose from or enter your own.", |
|
examples=examples) |
|
|
|
iface.launch() |
|
|