File size: 774 Bytes
8cc9f99
 
f7d0236
5e3abdf
8f69bd5
5e3abdf
d5e98e7
531cb56
ef8a5a9
d5e98e7
531cb56
8cc9f99
 
091877f
8cc9f99
d5e98e7
e79e5ca
8cc9f99
 
02113c8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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)