|
import gradio as gr |
|
from predict_utils import load_model, predict |
|
|
|
model, tokenizer = load_model() |
|
|
|
def classify(text): |
|
return predict(text, model, tokenizer) |
|
|
|
demo = gr.Interface( |
|
fn=classify, |
|
inputs=gr.Textbox(lines=4, placeholder="Masukkan teks cyberbullying..."), |
|
outputs=gr.Label(label="Prediksi Cyberbullying"), |
|
title="π¨ Deteksi Cyberbullying Berbahasa Indonesia", |
|
description="Model hybrid IndoBERT + BiLSTM untuk klasifikasi 10 kategori cyberbullying (multi-label).", |
|
examples=[ |
|
["Dasar kamu bodoh dan gak guna!"], |
|
["Kaum kalian harus dimusnahkan dari muka bumi."], |
|
["Kamu terlalu kurus, kayak tengkorak jalan!"] |
|
] |
|
) |
|
|
|
demo.launch() |