import gradio as gr

title = "Disaster tweets classification"

description = "Text classification using DistilBERT fine-tuned on the Kaggle disaster tweets dataset. \n Labels:\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)