Views
No views yet
indolem/indobert-base-uncased1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load model
5tokenizer = AutoTokenizer.from_pretrained("dansachs/indo-religiolect-bert")
6model = AutoModelForSequenceClassification.from_pretrained("dansachs/indo-religiolect-bert")
7
8# Predict
9text = "Allah adalah Tuhan yang Maha Esa"
10inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=128)
11outputs = model(**inputs)
12prediction = torch.argmax(outputs.logits, dim=-1).item()
13
14label_map = {0: 'Islam', 1: 'Catholic', 2: 'Protestant'}
15print(f"Prediction: {label_map[prediction]}")@misc{indo-religiolect-bert,
author = {Dan Sachs},
title = {Indo-Religiolect-BERT: Indonesian Religious Text Classifier},
year = {2025},
publisher = {Hugging Face},
howpublished = {\url{https://huggingface.co/dansachs/indo-religiolect-bert}}
}