Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("sasn59/gpt2-validator-balanced-20251104-1118")
4model = AutoModelForCausalLM.from_pretrained("sasn59/gpt2-validator-balanced-20251104-1118")
5
6# Generate text
7inputs = tokenizer("Hello, I am", return_tensors="pt")
8outputs = model.generate(**inputs, max_length=50)
9print(tokenizer.decode(outputs[0], skip_special_tokens=True))