Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load the model and tokenizer
4model_id = "your_username/chatbot-smart-home"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8# Example input
9input_text = "What smart lighting options can improve energy efficiency?"
10
11# Generate a response
12inputs = tokenizer(input_text, return_tensors="pt")
13outputs = model.generate(**inputs, max_length=256)
14response = tokenizer.decode(outputs[0], skip_special_tokens=True)
15
16print(response)