Spaces:
Running
Running
File size: 316 Bytes
15d9901 |
1 2 3 4 5 6 7 8 9 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
MODEL_NAME = "indobenchmark/indobert-base-p1"
MODEL_FINETUNED = "yondikavl/artour-spam-filter"
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
model = AutoModelForSequenceClassification.from_pretrained(MODEL_FINETUNED)
model.eval()
|