Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("omoplatapus/don")
4tokenizer = AutoTokenizer.from_pretrained("omoplatapus/don")
5
6# Format your input
7text = "Let's negotiate the price of this property. I'm offering $800,000 for it."
8
9# Generate response
10inputs = tokenizer(text, return_tensors="pt")
11outputs = model.generate(**inputs)
12response = tokenizer.decode(outputs[0])