Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("whis-22/bankai-recipe")
4tokenizer = AutoTokenizer.from_pretrained("whis-22/bankai-recipe")
5
6# Generate a recipe
7input_prompt = "Chocolate Chip Cookies Recipe:"
8input_ids = tokenizer.encode(input_prompt, return_tensors="pt")
9output = model.generate(input_ids, max_length=300, num_return_sequences=1)
10generated_recipe = tokenizer.decode(output[0], skip_special_tokens=True)
11print(generated_recipe)