hkayesh's picture
Update app.py
ef8a5a9
raw
history blame contribute delete
774 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"],
["Just happened a terrible car crash"],
["Heard about #earthquake in different cities"],
["I love my new Tesla!"],
["Wow social media really blew up after the Adele concert"]]
interface = gr.Interface.load("huggingface/hkayesh/twitter-disaster-nlp",
title=title,
description=description,
examples=examples,
theme="peach"
)
interface.launch(enable_queue=True)