Views
No views yet
1>>> from transformers import AutoModelForSequenceClassification, AutoTokenizer
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4model_name='rfahlevih/fine-tuned-cahya-distilbert-base-indonesia-reviews'
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSequenceClassification.from_pretrained(model_name)
7text = "Isi dengan sentimen apa saja."
8
9inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
10outputs = model(**inputs)
11logits = outputs.logits
12predicted_class = logits.argmax(dim=-1).item()
13
14sentiment = "Negative" if predicted_class == 0 else "Positive"
15
16print(f"Predicted class: {predicted_class}\nSentimen: {sentiment}")