Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4tokenizer = AutoTokenizer.from_pretrained("your-username/your-model-name", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained("your-username/your-model-name", trust_remote_code=True)
6
7# Generate text
8input_text = "The future of AI is"
9inputs = tokenizer(input_text, return_tensors="pt")
10outputs = model.generate(**inputs, max_length=50)
11generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(generated_text)