Spaces:
Running
Running
| import gradio as gr | |
| from transformers import pipeline | |
| model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english") | |
| def detect_fake_news(text): | |
| result = model(text)[0] | |
| label = result['label'] | |
| if label == "NEGATIVE": | |
| return "β Fake News" | |
| else: | |
| return "β Real News" | |
| interface = gr.Interface( | |
| fn=detect_fake_news, | |
| inputs=gr.Textbox(lines=7, placeholder="Enter news text here..."), | |
| outputs="text", | |
| title="Fake News Detector", | |
| description="A simple model to classify text as fake or real news." | |
| ) | |
| interface.launch() |