API berbasis Flask untuk klasifikasi komentar berbahasa Indonesia terkait judi online, menggunakan model IndoBERT yang telah di-fine-tune.
- Model klasifikasi berbasis IndoBERT
- Preprocessing teks bahasa Indonesia
- REST API sederhana (
/predict) - Mengembalikan label dan confidence score
- Model: fhru/indobert-judi-online-2
- Framework: TensorFlow + HuggingFace Transformers
- Arsitektur: IndoBERT (BERT pretrained untuk Bahasa Indonesia)
.
βββ app/
β βββ main.py
β βββ utils.py
β βββ requirements.txt
βββ Dockerfile
βββ .dockerignore
βββ README.md
-
Clone repository:
git clone https://github.com/HAJAR-Enterprise/ML-API.git cd ML_API -
Build Docker image:
docker build -t indobert-api . -
Jalankan image lokal:
docker run -p 8080:8080 indobert-api
Prediksi apakah komentar mengandung unsur judi online.
[
{
"commentId": "abc123",
"text": "main slot gacor gampang maxwin"
},
{
"commentId": "xyz789",
"text": "makasih infonya"
}
][
{
"commentId": "abc123",
"text": "main slot gacor gampang maxwin",
"label": "judi",
"confidence": 0.9845
},
{
"commentId": "xyz789",
"text": "makasih infonya",
"label": "normal",
"confidence": 0.9968
}
]