Views
No views yet
chef_gpt2_recipe_generator is a Causal Language Model designed to generate cooking recipes based on a provided list of ingredients. It is fine-tuned on a large corpus of structured recipes, learning the relationship between ingredients, quantities, and instructions.gpt2-medium architecture (355M parameters).INGREDIENTS: [list of items] \n INSTRUCTIONS: [step-by-step guide] <|endoftext|>.INGREDIENTS: tag.1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2
3model_name = "your_username/chef_gpt2_recipe_generator"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)
6
7chef = pipeline('text-generation', model=model, tokenizer=tokenizer, device=-1)
8
9# Define ingredients
10prompt = "INGREDIENTS: Chicken breast, garlic, soy sauce, honey, broccoli. \n INSTRUCTIONS:"
11
12output = chef(
13 prompt,
14 max_length=400,
15 num_return_sequences=1,
16 do_sample=True,
17 top_k=50,
18 top_p=0.92,
19 temperature=0.8,
20 pad_token_id=tokenizer.eos_token_id
21)
22
23print(output[0]['generated_text'])