import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification
loaded_model = AutoModelForSequenceClassification.from_pretrained("halilibr/dilbazlar-binary-disorder-detection-model-acc-92", num_labels=2)
tokenizer = AutoTokenizer.from_pretrained("halilibr/dilbazlar-binary-disorder-detection-model-acc-92")
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
loaded_model.to(device)
loaded_model.eval()
input_text = "kendimi kötü hissediyorum"
inputs = tokenizer(input_text, max_length=150, padding="max_length", truncation=True, return_tensors="pt")
inputs = {k: v.to(device) for k, v in inputs.items()}
print(inputs)
with torch.no_grad():
outputs = loaded_model(**inputs)
if hasattr(outputs, 'logits'):
preds = torch.argmax(outputs.logits, dim=-1)
else:
preds = torch.argmax(outputs[0], dim=-1)
prediction = preds.cpu().numpy()[0]
print("Predicted class:", prediction)