mab's picture
Update README.md
3228004

from transformers import AutoTokenizer, GPT2ForSequenceClassification

import torch

model_name= "mab/GPT2-German-sentiment"

tokenizer = AutoTokenizer.from_pretrained(model_name)

model = GPT2ForSequenceClassification.from_pretrained(model_name)

inputs = tokenizer("Mein Hund ist wunderschön", return_tensors="pt")

with torch.no_grad(): logits = model(**inputs).logits

predicted_class_id = logits.argmax().item()

class_label = model.config.id2label[predicted_class_id]

print(class_label)