hkayesh's picture
Update app.py
8f69bd5
raw
history blame
765 Bytes
import gradio as gr
title = "Disaster tweets classification"
description = "Text classification using DistilBERT fine-tuned on the Kaggle disaster tweets dataset. \n\nLabels:\n- LABEL_1 - About disaster \n- LABEL_0 - Not about disaster"
examples = [["Twelve feared killed in an air ambulance helicopter crash"],
["Emergency services called after 'strong' chemical smells"],
["I love my new Tesla!"],
["Wow social media really blew up after the Adele concert"],
["Are you OK?"]]
interface = gr.Interface.load("huggingface/hkayesh/twitter-disaster-nlp",
title=title,
description=description,
examples=examples,
theme="peach"
)
interface.launch(enable_queue=True)