christinacdl's picture
Update app.py
56bad1a
import gradio as gr
from transformers import pipeline
classifier = pipeline(task="text-classification", model="christinacdl/XLM_RoBERTa-Clickbait-Detection-new")
def text_classification(text):
result= classifier(text)
label = result[0]['label']
score = result[0]['score']
formatted_output = f"The text's label is {label}. I am {score*100:.2f}% sure!"
return formatted_output
examples=["Your Old Passport Will Be Replaced With Smart E-passport, Here Is Everything You Need To Know", "17 Things That Are Too Real For People Who Always Drop Their Phone", "Israeli Companies Seek a Worldwide Profile", "Venus Williams beats Marion Bartoli to triumph at Wimbledon"]
gradio_app = gr.Interface(fn=text_classification,
inputs= gr.Textbox(lines=2, label="Text", placeholder="Enter text here..."),
outputs=gr.Textbox(lines=2, label="Text Classification Result"),
title="Is It Clickbait?",
description="Enter a text to check if it is clickbait or not!",
examples=examples)
if __name__ == "__main__":
gradio_app.launch()