Views
No views yet
pip install transformers1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4# Load the model and tokenizer
5model_name = "byunal/electra-base-turkish-cased-stance"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9# Example text
10text = "Bu konu hakkında kesinlikle karşıyım."
11
12# Tokenize input
13inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
14
15# Perform prediction
16with torch.no_grad():
17 outputs = model(**inputs)
18
19# Get predicted stance
20predictions = torch.argmax(outputs.logits, dim=-1)
21stance_label = predictions.item()
22
23# Display result
24labels = ["Favor", "Against", "Neutral"]
25print(f"The stance is: {labels[stance_label]}")