import gradio as gr

title = "Disaster tweets classification"
description = "Text classification using DistilBERT fine-tuned on the Kaggle disaster tweets dataset"
examples = [["Twelve feared killed in an air ambulance helicopter crash"], 
            ["It's an apocalypse"], 
            ["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",
            description=description,
            examples=examples,
            theme="peach"
)

interface.launch()