import torch
from transformers import AutoTokenizer

model = torch.load("pytorch_model.bin")
tokenizer = AutoTokenizer.from_pretrained("alfiyahqthz/roberta-multiclass-das")

text = "I feel depressed"
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
logits = model(inputs['input_ids'], inputs['attention_mask'])
prediction = logits.argmax(dim=1).item()
# 0=Depression, 1=Anxiety, 2=Stress