Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3tokenizer = AutoTokenizer.from_pretrained("Asgar-Ali-T/yoda-chatbot")
4model = AutoModelForCausalLM.from_pretrained("Asgar-Ali-T/yoda-chatbot")
5
6# Example usage
7prompt = "Human: What is the Force?\nYoda:"
8inputs = tokenizer.encode(prompt, return_tensors="pt")
9outputs = model.generate(inputs, max_length=100, temperature=0.7)
10response = tokenizer.decode(outputs[0], skip_special_tokens=True)
11print(response)