File size: 229 Bytes
2f4b95c
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from transformers import pipeline

# Memuat model IndoBERT
classifier = pipeline("text-classification", model="indobenchmark/indobert-base-p1")

# Fungsi untuk memproses input
def classify_text(text):
    return classifier(text)